Skip to content

Commit

Permalink
change load and register function to classmethod
Browse files Browse the repository at this point in the history
  • Loading branch information
diegomarvid committed Mar 5, 2024
1 parent 74369b1 commit 93d304d
Showing 1 changed file with 2 additions and 7 deletions.
9 changes: 2 additions & 7 deletions pipeline_lib/core/pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,8 +61,8 @@ def auto_register_steps_from_package(cls, package_name):
):
cls.register_step(attribute)

@staticmethod
def load_and_register_custom_steps(custom_steps_path: str) -> None:
@classmethod
def load_and_register_custom_steps(cls, custom_steps_path: str) -> None:
"""
Dynamically loads and registers step classes found in the specified directory.
Expand All @@ -74,11 +74,6 @@ def load_and_register_custom_steps(custom_steps_path: str) -> None:
----------
custom_steps_path : str
The path to the directory containing custom step implementation files.
Returns
-------
NoReturn
This function does not return anything.
"""
Pipeline.logger.debug(f"Loading custom steps from: {custom_steps_path}")
for filename in os.listdir(custom_steps_path):
Expand Down

0 comments on commit 93d304d

Please sign in to comment.