diff --git a/datalab/datalab_session/data_operations/noop.py b/datalab/datalab_session/data_operations/noop.py index 731c8da..189c744 100644 --- 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 index 00878f4..0a901a5 100644 --- 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)