Skip to content

Commit

Permalink
Reorder imports
Browse files Browse the repository at this point in the history
Signed-off-by: Ryan Wolf <[email protected]>
  • Loading branch information
ryantwolf committed Apr 1, 2024
1 parent fc167a6 commit 06dbde3
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion nemo_curator/modules/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,14 +13,17 @@
# limitations under the License.

from .add_id import AddId
from .distributed_data_classifier import DomainClassifier, QualityClassifier
from .exact_dedup import ExactDuplicates
from .filter import Filter, Score, ScoreFilter
from .fuzzy_dedup import LSH, MinHash
from .meta import Sequential
from .modify import Modify
from .task import TaskDecontamination

# This import must come after all imports that require cugraph
# See this issue: https://github.com/rapidsai/cugraph/issues/2718
from .distributed_data_classifier import DomainClassifier, QualityClassifier

__all__ = [
"DomainClassifier",
"ExactDuplicates",
Expand Down

0 comments on commit 06dbde3

Please sign in to comment.