module error

KoslickiLab · Nov 14, 2023 · 5c00a9c · 5c00a9c
1 parent 423e923
commit 5c00a9c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/srcs/cli.py b/srcs/cli.py
@@ -1,8 +1,9 @@
 import argparse
-import run_YACHT as run_YACHT
-import make_training_data_from_sketches as make_training_data
 
-def main():
+from srcs import run_YACHT
+
+
+def main(make_training_data=None):
     parser = argparse.ArgumentParser(prog='yacht')
     subparsers = parser.add_subparsers(dest='command')