Browse Source

fixed some import for convenience

Dimitri Korsch 5 năm trước cách đây
mục cha
commit
5ebc6a2967
1 tập tin đã thay đổi với 9 bổ sung2 xóa
  1. 9 2
      cvfinetune/parser/__init__.py

+ 9 - 2
cvfinetune/parser/__init__.py

@@ -1,8 +1,15 @@
-from cvfinetune.parser.base import FineTuneParser, default_factory
+from cvfinetune.parser.base import FineTuneParser
+from cvfinetune.parser.base import add_dataset_args
+from cvfinetune.parser.base import add_model_args
+from cvfinetune.parser.base import add_training_args
+from cvfinetune.parser.base import default_factory
 
 
 __all__ = [
+	"add_dataset_args",
+	"add_model_args",
+	"add_training_args",
+	"default_factory",
 	"FineTuneParser",
-	"default_factory"
 ]