ocean-runner 0.2.12__py3-none-any.whl → 0.2.14__py3-none-any.whl

ocean_runner/config.py

@@ -2,7 +2,7 @@ import os
 from dataclasses import asdict, dataclass, field
 from logging import Logger
 from pathlib import Path
-from typing import Callable, Iterable, TypeVar
+from typing import Iterable, TypeVar
 
 T = TypeVar("T")
 
@@ -43,9 +43,6 @@ class Config:
     custom_input: T | None = None
     """Algorithm's custom input types, must be a dataclass_json"""
 
-    error_callback: Callable[[Exception], None] = None
-    """Callback to execute upon exceptions"""
-
     logger: Logger | None = None
     """Logger to use in the algorithm"""
 
@@ -54,7 +51,5 @@ class Config:
     )
     """Paths that should be included so the code executes correctly"""
 
-    environment: Environment = field(
-        default_factory=lambda: Environment(),
-    )
+    environment: Environment = field(default_factory=lambda: Environment())
     """Mock of environment data"""

ocean_runner/runner.py

@@ -9,49 +9,52 @@ from oceanprotocol_job_details import JobDetails
 
 from ocean_runner.config import Config
 
-JobDetailsT = TypeVar(
-    "JobDetailsT",
-)
+JobDetailsT = TypeVar("JobDetailsT")
 ResultT = TypeVar("ResultT")
 
 
-def default_error_callback(_: Algorithm, e: Exception) -> None:
+def default_error_callback(_, e: Exception) -> None:
     raise e
 
 
 def default_validation(algorithm: Algorithm) -> None:
     algorithm.logger.info("Validating input using default validation")
-
     assert algorithm.job_details.ddos, "DDOs missing"
     assert algorithm.job_details.files, "Files missing"
 
 
 def default_save(*, result: ResultT, base: Path, algorithm: Algorithm) -> None:
     algorithm.logger.info("Saving results using default save")
-
     with open(base / "result.txt", "w+") as f:
         f.write(str(result))
 
 
 @dataclass
 class Algorithm(Generic[JobDetailsT, ResultT]):
+    """
+    A configurable algorithm runner that behaves like a FastAPI app:
+      - You register `validate`, `run`, and `save_results` via decorators.
+      - You execute the full pipeline by calling `app()`.
+    """
 
     config: InitVar[Config | None] = None
-
-    # Load from config
     logger: Logger = field(init=False)
-
     _job_details: JobDetails[JobDetailsT] = field(init=False)
     _result: ResultT | None = field(default=None, init=False)
 
-    error_callback = default_error_callback
+    error_callback: Callable[[Algorithm, Exception], None] = default_error_callback
+
+    # Decorator-registered callbacks
+    _validate_fn: Callable[[Algorithm], None] | None = field(default=None, init=False)
+    _run_fn: Callable[[Algorithm], ResultT] | None = field(default=None, init=False)
+    _save_fn: Callable[[ResultT, Path, Algorithm], None] | None = field(
+        default=None, init=False
+    )
 
     def __post_init__(self, config: Config | None) -> None:
         config: Config = config or Config()
 
-        if config.error_callback:
-            self.error_callback = config.error_callback
-
+        # Configure logger
         if config.logger:
             self.logger = config.logger
         else:
@@ -62,18 +65,20 @@ class Algorithm(Generic[JobDetailsT, ResultT]):
                 format="%(asctime)s | %(levelname)-8s | %(name)s | %(message)s",
                 datefmt="%Y-%m-%d %H:%M:%S",
             )
-
             self.logger = logging.getLogger("ocean_runner")
 
+        # Normalize base_dir
         if isinstance(config.environment.base_dir, str):
             config.environment.base_dir = Path(config.environment.base_dir)
 
+        # Extend sys.path for custom imports
         if config.source_paths:
             import sys
 
             sys.path.extend([str(path.absolute()) for path in config.source_paths])
             self.logger.debug(f"Added [{len(config.source_paths)}] entries to PATH")
 
+        # Load job details
         self._job_details = JobDetails.load(
             _type=config.custom_input,
             base_dir=config.environment.base_dir,
@@ -85,6 +90,8 @@ class Algorithm(Generic[JobDetailsT, ResultT]):
         self.logger.info("Loaded JobDetails")
         self.logger.debug(asdict(self.job_details))
 
+        self.config = config
+
     class Error(RuntimeError): ...
 
     @property
@@ -99,45 +106,67 @@ class Algorithm(Generic[JobDetailsT, ResultT]):
             raise Algorithm.Error("Result missing, run the algorithm first")
         return self._result
 
-    def validate(
-        self,
-        callback: Callable[[Self], None] = default_validation,
-    ) -> Self:
-        self.logger.info("Validating instance...")
-        try:
-            callback(self)
-        except Exception as e:
-            self.error_callback(e)
+    # ---------------------------
+    # Decorators (FastAPI-style)
+    # ---------------------------
 
-        return self
+    def validate(self, fn: Callable[[Self], None]) -> Callable[[Self], None]:
+        self._validate_fn = fn
+        return fn
 
-    def run(
-        self,
-        callable: Callable[[Self], ResultT],
-    ) -> Self:
-        self.logger.info("Running algorithm...")
-        try:
-            self._result = callable(self)
-        except Exception as e:
-            self.error_callback(e)
+    def run(self, fn: Callable[[Self], ResultT]) -> Callable[[Self], ResultT]:
+        self._run_fn = fn
+        return fn
+
+    def save_results(self, fn: Callable[[ResultT, Path, Algorithm], None]) -> Callable:
+        self._save_fn = fn
+        return fn
+
+    def on_error(self, fn: Callable[[Algorithm, Exception], None]) -> Callable:
+        self.error_callback = fn
+        return fn
 
-        return self
+    # ---------------------------
+    # Execution Pipeline
+    # ---------------------------
 
-    def save_results(
-        self,
-        callable: Callable[[ResultT, Path, Algorithm], None] = default_save,
-        *,
-        override_path: Path | None = None,
-    ) -> None:
-        self.logger.info("Saving results...")
+    def __call__(self) -> ResultT | None:
+        """Executes the algorithm pipeline: validate → run → save_results."""
         try:
-            callable(
-                results=self.result,
-                base_path=override_path or self.job_details.paths.outputs,
-                algorithm=self,
-            )
-        except Exception as e:
-            self.error_callback(e)
+            # Validation step
+            if self._validate_fn:
+                self.logger.info("Running custom validation...")
+                self._validate_fn(self)
+            else:
+                self.logger.info("Running default validation...")
+                default_validation(self)
+
+            # Run step
+            if self._run_fn:
+                self.logger.info("Running algorithm...")
+                self._result = self._run_fn(self)
+            else:
+                self.logger.warning("No run() function defined. Skipping execution.")
+                self._result = None
+
+            # Save step
+            if self._save_fn:
+                self.logger.info("Saving results...")
+                self._save_fn(
+                    self._result,
+                    self.job_details.paths.outputs,
+                    self,
+                )
+            else:
+                self.logger.info("No save_results() defined. Using default.")
+                default_save(
+                    result=self._result,
+                    base=self.job_details.paths.outputs,
+                    algorithm=self,
+                )
 
+        except Exception as e:
+            self.logger.exception("Error during algorithm execution")
+            self.error_callback(self, e)
 
-__all__ = [Algorithm]
+        return self._result

{ocean_runner-0.2.12.dist-info → ocean_runner-0.2.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ocean-runner
-Version: 0.2.12
+Version: 0.2.14
 Summary: A fluent API for OceanProtocol algorithms
 Project-URL: Homepage, https://github.com/AgrospAI/ocean-runner
 Project-URL: Issues, https://github.com/AgrospAI/ocean-runner/issues

ocean_runner-0.2.14.dist-info/RECORD

@@ -0,0 +1,7 @@
+ocean_runner/__init__.py,sha256=awAmE6kZhuwcrD3gT7qFZArdhiuzW-EFTA6tGKhw06k,138
+ocean_runner/config.py,sha256=gyyUotPJ7n8wPPdsJZIBUT4zBlkoNbhV876JDTdPNsY,1398
+ocean_runner/runner.py,sha256=AM4GlgpGGOSLa_TfO8rAdWmTGTPTZoTT84LNtUswTXw,5762
+ocean_runner-0.2.14.dist-info/METADATA,sha256=bIdyxazZ4mWCOUW7JULJaykDgDCd03yZ91CWG4BbsT8,5919
+ocean_runner-0.2.14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+ocean_runner-0.2.14.dist-info/licenses/LICENSE,sha256=_B25KqK4amoADWkMN150tnZFm_Fy7VvZpvIC8ZydWdI,1053
+ocean_runner-0.2.14.dist-info/RECORD,,

ocean_runner-0.2.12.dist-info/RECORD

@@ -1,7 +0,0 @@
-ocean_runner/__init__.py,sha256=awAmE6kZhuwcrD3gT7qFZArdhiuzW-EFTA6tGKhw06k,138
-ocean_runner/config.py,sha256=EgloiUTLBvca4B_pr2eNa8kHYfsyVE73ZxszLCbiDYY,1525
-ocean_runner/runner.py,sha256=LXkVNYsqELsXciI8T8RWtfbNpI-_kUAtW1aQzsnOAfU,4075
-ocean_runner-0.2.12.dist-info/METADATA,sha256=k9dRV2nI-i3_ZJPETo2DnuDzasRqx_YZhFe3pVy-20M,5919
-ocean_runner-0.2.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-ocean_runner-0.2.12.dist-info/licenses/LICENSE,sha256=_B25KqK4amoADWkMN150tnZFm_Fy7VvZpvIC8ZydWdI,1053
-ocean_runner-0.2.12.dist-info/RECORD,,