Update launch.json and main.py

supervisely-ecosystem · Aug 14, 2024 · d4b8dff · d4b8dff
1 parent 3e010b5
commit d4b8dff
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 6 deletions.
diff --git a/.vscode/launch.json b/.vscode/launch.json
@@ -3,7 +3,7 @@
     "configurations": [
         {
             "name": "Python: Current File",
-            "type": "python",
+            "type": "debugpy",
             "request": "launch",
             "program": "src/main.py",
             "console": "integratedTerminal",

diff --git a/src/main.py b/src/main.py
@@ -2,12 +2,13 @@
 
 import supervisely as sly
 
-import src.globals as g
 import src.functions as f
+import src.globals as g
 import src.workflow as w
 
 api = None
 
+
 class MyExport(sly.app.Export):
     def process(self, context: sly.app.Export.Context):
         global api
@@ -52,9 +53,13 @@ def process(self, context: sly.app.Export.Context):
         progress = sly.Progress("Transformation ...", images_count)
         for ds in datasets:
             train_info, val_info = f.process_images(
-                api, meta, ds,
-                class_names, progress,
-                dir_names, skipped,
+                api,
+                meta,
+                ds,
+                class_names,
+                progress,
+                dir_names,
+                skipped,
                 max_kpts_count,
             )
             train_ids, train_img_paths, train_anns = train_info

diff --git a/src/workflow.py b/src/workflow.py
@@ -12,7 +12,6 @@ def workflow_input(api: sly.Api, project_id: int):
 
 def workflow_output(api: sly.Api, file: Union[int, sly.api.file_api.FileInfo]):
     try:
-        api.file.get
         if isinstance(file, int):
             file = api.file.get_info_by_id(file)
         relation_settings = sly.WorkflowSettings(