add decorator to __all__ (40fab6cc) · Commits · csst-pipeline / csst_common

csst_common/init.py

+6 −1

Original line number	Diff line number	Diff line
		@@ -17,4 +17,9 @@ from .status import CsstResult, CsstStatus

		__version__ = "0.0.2"

		__all__ = ["Pipeline", "File", "CsstResult"]
		__all__ = [
		"Pipeline",
		"File",
		"CsstResult",
		"parameterized_module_decorator",
		]

+16 −0

Original line number	Diff line number	Diff line
		@@ -32,6 +32,22 @@ class ModuleResult(NamedTuple):
		def parameterized_module_decorator(
		logger: Optional[logging.Logger] = None,
		) -> Callable:
		"""
		Parameterized module decorator.

		This is designed for wrapping modules.

		Parameters
		----------
		logger : Optional[logging.Logger]
		The logger.

		Returns
		-------
		Callable
		A wrapped module.
		"""

		# use default logger
		if logger is None:
		logger = get_logger()