Quick update to make no-op do something

LCOGT · Feb 7, 2024 · 83d65a3 · 83d65a3
1 parent 3b49c88
commit 83d65a3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/datalab/datalab_session/data_operations/noop.py b/datalab/datalab_session/data_operations/noop.py
@@ -41,4 +41,4 @@ def wizard_description():
         }
 
     def operate(self, input_data):
-        pass
+        print("No-op triggered!")
diff --git a/datalab/datalab_session/tasks.py b/datalab/datalab_session/tasks.py
@@ -1,13 +1,13 @@
 import dramatiq
+
 from datalab.datalab_session.data_operations.data_operation import available_operations
 
 #TODO: Perhaps define a pipeline that can take the output of one data operation and upload to a s3 bucket, indicate success, etc...
 
 @dramatiq.actor()
 def execute_data_operation(data_operation_name: str, input_data: dict):
-    clz = available_operations().get(data_operation_name)
-    if clz is None:
+    operation_class = available_operations().get(data_operation_name)
+    if operation_class is None:
         raise NotImplementedError("Operation not implemented!")
     else:
-        instance = clz()
-        instance.operate(input_data)
+        operation_class().operate(input_data)
-Original file line number
+Diff line change
@@ Expand Up / @@ -41,4 +41,4 @@ def wizard_description(): @@
             }
         def operate(self, input_data):
-            pass
+            print("No-op triggered!")