Skip to content

Commit

Permalink
some refactoring changes to satisfy mypy.
Browse files Browse the repository at this point in the history
  • Loading branch information
J-Dymond committed Nov 26, 2024
1 parent de365f3 commit 68c5268
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions src/arc_spice/variational_pipelines/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,11 +15,20 @@ class DummyPipeline:
"""

def __init__(self, model_name):
"""
Gives the dummy pipeline the required attributes for the method definitions
Args:
model_name: name of the pipeline that is being given a dummy
"""
self.model = model_name

def __call__(self, *args, **kwargs):
"""
Needs to be defined in subclass
Raises:
NotImplementedError: when called to prevent base class being used
"""
error_msg = (
f"{self.model} cannot be called directly and needs to be"
Expand Down

0 comments on commit 68c5268

Please sign in to comment.