Remove unneeded check

Kiln-AI · Nov 22, 2024 · 5e42d07 · 5e42d07
1 parent a95e613
commit 5e42d07
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/libs/core/kiln_ai/adapters/fine_tune/__init__.py b/libs/core/kiln_ai/adapters/fine_tune/__init__.py
@@ -4,10 +4,6 @@
 A set of classes for fine-tuning models.
 """
 
-import sys
+from . import dataset_split
 
-# Avoid circular import since we use datamodel in some tests
-if "pytest" not in sys.modules:
-    from . import dataset_split
-
-    __all__ = ["dataset_split"]
+__all__ = ["dataset_split"]