Remove _generated from some operators

TODO: reintroduce it in a more abstract way Signed-off-by: Krzysztof Lecki <klecki@nvidia.com>
NVIDIA · Oct 23, 2023 · c3bea75 · c3bea75
1 parent 65c4da7
commit c3bea75
Show file tree

Hide file tree

Showing 4 changed files with 0 additions and 7 deletions.
diff --git a/dali/python/nvidia/dali/external_source.py b/dali/python/nvidia/dali/external_source.py
@@ -830,7 +830,6 @@ def __call__(self, *, source=None, cycle=None, name=None, layout=None, dtype=Non
 
     __doc__ += _args_doc
     __call__.__doc__ += _args_doc
-    _generated = False
 
 
 def _is_external_source_with_callback(op_instance):
@@ -911,4 +910,3 @@ def _external_source(source=None, num_outputs=None, *, cycle=None, name=None, de
 
 
 external_source.__doc__ += ExternalSource._args_doc
-external_source._generated = False
diff --git a/dali/python/nvidia/dali/ops/_operators/python_function.py b/dali/python/nvidia/dali/ops/_operators/python_function.py
@@ -99,7 +99,6 @@ def _dlpack_from_array(array):
 
 class PythonFunction(PythonFunctionBase):
     schema_name = "PythonFunction"
-    _generated = False
     _registry.register_cpu_op('PythonFunction')
     _registry.register_gpu_op('PythonFunction')
 
@@ -217,7 +216,6 @@ def func(*ts):
 
 class DLTensorPythonFunction(PythonFunctionBase):
     schema_name = "DLTensorPythonFunction"
-    _generated = False
     _registry.register_cpu_op('DLTensorPythonFunction')
     _registry.register_gpu_op('DLTensorPythonFunction')
 

diff --git a/dali/python/nvidia/dali/ops/_operators/tfrecord.py b/dali/python/nvidia/dali/ops/_operators/tfrecord.py
@@ -101,10 +101,8 @@ def __call__(self, *inputs, **kwargs):
 
 
 class TFRecordReader(_TFRecordReaderImpl, metaclass=ops._DaliOperatorMeta):
-    _generated = False
     _internal_schema_name = '_TFRecordReader'
 
 
 class TFRecord(_TFRecordReaderImpl, metaclass=ops._DaliOperatorMeta):
-    _generated = False
     _internal_schema_name = 'readers___TFRecord'
diff --git a/dali/python/nvidia/dali/plugin/numba/experimental.py b/dali/python/nvidia/dali/plugin/numba/experimental.py
@@ -86,7 +86,6 @@ def _get_shape_view(shapes_ptr, ndims_ptr, num_dims, num_samples):
 
 class NumbaFunction(metaclass=ops._DaliOperatorMeta):
     schema_name = 'NumbaFunction'
-    _generated = False
     ops.register_cpu_op('NumbaFunction')
     ops.register_gpu_op('NumbaFunction')