[Fix] Small fixes to account for changes to RegistryMixin (#1558)

Author: dbogunowicz

src/deepsparse/operators/registry.py

@@ -33,14 +33,10 @@ class OperatorRegistry(RegistryMixin):
     """
 
     @classmethod
-    def register_value(cls, operator, name):
+    def register_value(cls, operator, name, alias):
         from deepsparse.operators import Operator
 
-        if not isinstance(name, list):
-            name = [name]
-
-        for task_name in name:
-            register(Operator, operator, task_name, require_subclass=True)
+        register(Operator, operator, name, alias, require_subclass=True)
 
         return operator
 

src/deepsparse/transformers/pipelines/code_generation.py

@@ -20,7 +20,7 @@
 __all__ = ["CodeGenerationPipeline"]
 
 
-@OperatorRegistry.register(name=["code_generation", "code_gen", "codegen"])
+@OperatorRegistry.register(name="code_generation", alias=["code_gen", "codegen"])
 class CodeGenerationPipeline(TextGenerationPipeline):
     """
     Subclass of text generation pipeline to support any defaults or

src/deepsparse/transformers/pipelines/text_generation/pipeline.py

@@ -54,7 +54,7 @@
 _LOGGER = logging.getLogger(__name__)
 
 
-@OperatorRegistry.register(name=["text_generation", "opt", "mpt", "llama"])
+@OperatorRegistry.register(name="text_generation", alias=["opt", "mpt", "llama"])
 class TextGenerationPipeline(Pipeline):
     DEFAULT_SEQUENCE_LENGTH = 1024
 

tests/deepsparse/evaluation/test_registry.py

@@ -52,7 +52,7 @@
         ),
         (
             "image_classification",
-            "zoo:cv/classification/resnet_v1-50_2x/pytorch/sparseml/imagenet/base-none",
+            "zoo:cv/classification/resnet_v1-50/pytorch/sparseml/imagenet/base-none",
             [
                 "-c",
                 "tests/test_data/pipeline_bench_config.json",
@@ -66,7 +66,7 @@
         ),
         (
             "image_classification",
-            "zoo:cv/classification/resnet_v1-50_2x/pytorch/sparseml/imagenet/base-none",
+            "zoo:cv/classification/resnet_v1-50/pytorch/sparseml/imagenet/base-none",
             [],
         ),
         (