From 63efe03106231ada2d2ea3ddd7f4090a71f260f5 Mon Sep 17 00:00:00 2001 From: Ryan Wolf Date: Mon, 1 Apr 2024 16:57:47 -0700 Subject: [PATCH] Clarify comment Signed-off-by: Ryan Wolf --- nemo_curator/modules/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/nemo_curator/modules/__init__.py b/nemo_curator/modules/__init__.py index 631c08c02..d845441f3 100644 --- a/nemo_curator/modules/__init__.py +++ b/nemo_curator/modules/__init__.py @@ -20,7 +20,8 @@ from .modify import Modify from .task import TaskDecontamination -# This import must come after all imports that require cugraph +# Pytorch related imports must come after all imports that require cugraph, +# because of context cleanup issues b/w pytorch and cugraph # See this issue: https://github.com/rapidsai/cugraph/issues/2718 from .distributed_data_classifier import DomainClassifier, QualityClassifier