Skip to content

Commit 93d304d

Browse files
committed
change load and register function to classmethod
1 parent 74369b1 commit 93d304d

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

pipeline_lib/core/pipeline.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -61,8 +61,8 @@ def auto_register_steps_from_package(cls, package_name):
6161
):
6262
cls.register_step(attribute)
6363

64-
@staticmethod
65-
def load_and_register_custom_steps(custom_steps_path: str) -> None:
64+
@classmethod
65+
def load_and_register_custom_steps(cls, custom_steps_path: str) -> None:
6666
"""
6767
Dynamically loads and registers step classes found in the specified directory.
6868
@@ -74,11 +74,6 @@ def load_and_register_custom_steps(custom_steps_path: str) -> None:
7474
----------
7575
custom_steps_path : str
7676
The path to the directory containing custom step implementation files.
77-
78-
Returns
79-
-------
80-
NoReturn
81-
This function does not return anything.
8277
"""
8378
Pipeline.logger.debug(f"Loading custom steps from: {custom_steps_path}")
8479
for filename in os.listdir(custom_steps_path):

0 commit comments

Comments
 (0)