feat: also allow passing an imported module to Tesseract.from_tesseract_api (#130)

#### Relevant issue or PR
n/a

#### Description of changes
Also allow passing an imported module to `Tesseract.from_tesseract_api`.
For example, this allows users to monkeypatch or construct modules at
runtime without having to write them to a file.

#### Testing done
new test on CI

#### License

- [x] By submitting this pull request, I confirm that my contribution is
made under the terms of the [Apache 2.0
license](https://pasteurlabs.github.io/tesseract/LICENSE).
- [x] I sign the Developer Certificate of Origin below by adding my name
and email address to the `Signed-off-by` line.

<details>
<summary><b>Developer Certificate of Origin</b></summary>

```text
Developer Certificate of Origin
Version 1.1

Copyright (C) 2004, 2006 The Linux Foundation and its contributors.

Everyone is permitted to copy and distribute verbatim copies of this
license document, but changing it is not allowed.


Developer's Certificate of Origin 1.1

By making a contribution to this project, I certify that:

(a) The contribution was created in whole or in part by me and I
    have the right to submit it under the open source license
    indicated in the file; or

(b) The contribution is based upon previous work that, to the best
    of my knowledge, is covered under an appropriate open source
    license and I have the right under that license to submit that
    work with modifications, whether created in whole or in part
    by me, under the same open source license (unless I am
    permitted to submit under a different license), as indicated
    in the file; or

(c) The contribution was provided directly to me by some other
    person who certified (a), (b) or (c) and I have not modified
    it.

(d) I understand and agree that this project and the contribution
    are public and that a record of the contribution (including all
    personal information I submit with it, including my sign-off) is
    maintained indefinitely and may be redistributed consistent with
    this project or the open source license(s) involved.
```

</details>

Signed-off-by: Dion Häfner <dion.haefner@simulation.science>

Author: dionhaefner

tesseract_core/sdk/tesseract.py

@@ -9,6 +9,7 @@
 from dataclasses import dataclass
 from functools import cached_property, wraps
 from pathlib import Path
+from types import ModuleType
 from typing import Any, Callable
 from urllib.parse import urlparse, urlunparse
 
@@ -127,26 +128,43 @@ def from_image(
     @classmethod
     def from_tesseract_api(
         cls,
-        tesseract_api_path: str | Path,
+        tesseract_api: str | Path | ModuleType,
     ) -> Tesseract:
-        """Create a Tesseract instance from a Tesseract API path.
+        """Create a Tesseract instance from a Tesseract API module.
 
         Warning: This does not use a containerized Tesseract, but rather
         imports the Tesseract API directly. This is useful for debugging,
         but requires a matching runtime environment + all dependencies to be
         installed locally.
 
         Args:
-            tesseract_api_path: Path to the `tesseract_api.py` file.
+            tesseract_api: Path to the `tesseract_api.py` file, or an
+                already imported Tesseract API module.
 
         Returns:
             A Tesseract instance.
         """
+        if isinstance(tesseract_api, str | Path):
+            from tesseract_core.runtime.core import load_module_from_path
+
+            tesseract_api_path = Path(tesseract_api).resolve(strict=True)
+            if not tesseract_api_path.is_file():
+                raise RuntimeError(
+                    f"Tesseract API path {tesseract_api_path} is not a file."
+                )
+
+            try:
+                tesseract_api = load_module_from_path(tesseract_api_path)
+            except ImportError as ex:
+                raise RuntimeError(
+                    f"Cannot load Tesseract API from {tesseract_api_path}"
+                ) from ex
+
         obj = cls.__new__(cls)
         obj._spawn_config = None
         obj._serve_context = None
         obj._lastlog = None
-        obj._client = LocalClient(tesseract_api_path)
+        obj._client = LocalClient(tesseract_api)
         return obj
 
     def __enter__(self) -> Tesseract:
@@ -584,23 +602,10 @@ def run_tesseract(self, endpoint: str, payload: dict | None = None) -> dict:
 class LocalClient:
     """Local Client for Tesseracts."""
 
-    def __init__(self, tesseract_api_path: str | Path) -> None:
-        from tesseract_core.runtime.core import create_endpoints, load_module_from_path
+    def __init__(self, tesseract_api: ModuleType) -> None:
+        from tesseract_core.runtime.core import create_endpoints
         from tesseract_core.runtime.serve import create_rest_api
 
-        tesseract_api_path = Path(tesseract_api_path).resolve(strict=True)
-        if not tesseract_api_path.is_file():
-            raise RuntimeError(
-                f"Tesseract API path {tesseract_api_path} is not a file."
-            )
-
-        try:
-            tesseract_api = load_module_from_path(tesseract_api_path)
-        except ImportError as ex:
-            raise RuntimeError(
-                f"Cannot load Tesseract API from {tesseract_api_path}"
-            ) from ex
-
         self._endpoints = {
             func.__name__: func for func in create_endpoints(tesseract_api)
         }

tests/sdk_tests/test_tesseract.py

@@ -45,6 +45,28 @@ def test_Tesseract_init():
         pass
 
 
+def test_Tesseract_from_tesseract_api(dummy_tesseract_location, dummy_tesseract_module):
+    all_endpoints = [
+        "apply",
+        "jacobian",
+        "jacobian_vector_product",
+        "vector_jacobian_product",
+        "health",
+        "input_schema",
+        "output_schema",
+        "abstract_eval",
+    ]
+
+    t = Tesseract.from_tesseract_api(dummy_tesseract_location / "tesseract_api.py")
+    endpoints = t.available_endpoints
+    assert endpoints == all_endpoints
+
+    # should also work when importing the module
+    t = Tesseract.from_tesseract_api(dummy_tesseract_module)
+    endpoints = t.available_endpoints
+    assert endpoints == all_endpoints
+
+
 def test_Tesseract_from_image(mock_serving, mock_clients):
     # Object is built and has the correct attributes set
     t = Tesseract.from_image("sometesseract:0.2.3", volumes=["/my/files"], gpus=["all"])