nebu 0.1.6__py3-none-any.whl → 0.1.8__py3-none-any.whl

nebu/containers/container.py

@@ -26,7 +26,7 @@ class Container:
         self,
         name: str,
         image: str,
-        namespace: str = "default",
+        namespace: Optional[str] = None,
         platform: Optional[str] = None,
         env: Optional[List[V1EnvVar]] = None,
         command: Optional[str] = None,
@@ -208,8 +208,8 @@ class Container:
 
         # Save constructor params to `self` for reference, like you do in ReplayBuffer.
         self.kind = "Container"
-        self.namespace = namespace
-        self.name = name
+        self.namespace = self.container.metadata.namespace
+        self.name = self.container.metadata.name
         self.platform = platform
         self.metadata = meta_request
         self.image = image
@@ -290,7 +290,7 @@ class Container:
     def load(
         cls,
         name: str,
-        namespace: str = "default",
+        namespace: Optional[str] = None,
         config: Optional[GlobalConfig] = None,
     ):
         """

nebu/containers/decorator.py

@@ -7,13 +7,12 @@ from typing import Any, Callable, List, Optional
 import requests
 
 from nebu.containers.container import Container
-from nebu.containers.models import V1ContainerRequest, V1EnvVar, V1ResourceMetaRequest
 
 
 def container(
     image: str,
     name: Optional[str] = None,
-    namespace: str = "default",
+    namespace: Optional[str] = None,
     accelerators: Optional[List[str]] = None,
     platform: str = "runpod",
     python_cmd: str = "python",
@@ -84,91 +83,3 @@ def container(
         return wrapper
 
     return decorator
-
-
-def on_feedback(
-    human: Human,
-    accelerators: Optional[List[str]] = None,
-    platform: str = "runpod",
-    python_cmd: str = "python",
-    timeout: Optional[str] = None,
-    env: Optional[List[V1EnvVar]] = None,
-):
-    def decorator(func: Callable):
-        nonlocal name
-        if name is None:
-            name = func.__name__
-
-        # Get function source code
-        func_code = inspect.getsource(func)
-
-        command = """
-
-"""
-
-        # Create the container request
-        container_request = V1ContainerRequest(
-            kind="Container",
-            platform=platform,
-            metadata=V1ResourceMetaRequest(
-                name=name,
-                namespace=namespace,
-            ),
-            image=image,
-            env=env,
-            command=f"{python_cmd} -m nebu.containers.server",
-            accelerators=accelerators,
-            timeout=timeout,
-            proxy_port=8080,
-            restart="Never",  # Jobs should not restart
-        )
-
-        def run(*args: Any, **kwargs: Any):
-            # Create a container from the request
-            cont = Container.from_request(container_request)
-
-            # Wait for container to be running
-            while (
-                cont.status
-                and cont.status.status
-                and cont.status.status.lower() != "running"
-            ):
-                print(f"Job '{cont.metadata.name}' not running yet; waiting...")
-                time.sleep(1)
-
-            # Serialize arguments using pickle for complex objects
-            serialized_args = base64.b64encode(pickle.dumps(args)).decode("utf-8")
-            serialized_kwargs = base64.b64encode(pickle.dumps(kwargs)).decode("utf-8")
-
-            # Prepare payload
-            payload = {
-                "function_code": func_code,
-                "args": serialized_args,
-                "kwargs": serialized_kwargs,
-            }
-
-            # Get container URL
-            container_url = (
-                cont.status.tailnet_url
-                if cont.status and hasattr(cont.status, "tailnet_url")
-                else "http://localhost:8080"
-            )
-
-            # Send to container and get result
-            response = requests.post(f"{container_url}/execute", json=payload)
-
-            if response.status_code != 200:
-                raise RuntimeError(f"Function execution failed: {response.text}")
-
-            # Deserialize the result
-            serialized_result = response.json()["result"]
-            result = pickle.loads(base64.b64decode(serialized_result))
-
-            return result
-
-        # Attach the run method to the container request
-        container_request.run = run  # type: ignore
-
-        return container_request
-
-    return decorator

{nebu-0.1.6.dist-info → nebu-0.1.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nebu
-Version: 0.1.6
+Version: 0.1.8
 Summary: A globally distributed container runtime
 Requires-Python: >=3.10.14
 Description-Content-Type: text/markdown

{nebu-0.1.6.dist-info → nebu-0.1.8.dist-info}/RECORD

@@ -1,16 +1,16 @@
 nebu/__init__.py,sha256=EbdC8ZKnRTt6jkX0WN0p1pnaDEzb2InqZ1r8QZWzph0,195
 nebu/config.py,sha256=XBY7uKgcJX9d1HGxqqpx87o_9DuF3maUlUnKkcpUrKU,4565
 nebu/meta.py,sha256=CzFHMND9seuewzq9zNNx9WTr6JvrCBExe7BLqDSr7lM,745
-nebu/containers/container.py,sha256=JbqdzjsagdlPmB8J_IueVqWuLixPAQ6RW9PRVlnRiqE,13444
-nebu/containers/decorator.py,sha256=sZQ4ZwqQk_jLDJ-n9P6N0MlxvWL9Ac7eCPIHVxYq58c,5576
+nebu/containers/container.py,sha256=yb7KaPTVXnEEAlrpdlUi4HNqF6P7z9bmwAILGlq6iqU,13502
+nebu/containers/decorator.py,sha256=qiM7hbHne9MhSp1gDgX5z5bimsXr_YPjTIZoe09dwr4,2741
 nebu/containers/models.py,sha256=_d6BS6puoVWvyHhWX-74WFHJSOE8WJaFt2zGMTm9EEA,6782
 nebu/containers/server.py,sha256=yFa2Y9PzBn59E1HftKiv0iapPonli2rbGAiU6r-wwe0,2513
 nebu/processors/models.py,sha256=6XSw4iM77XYJf6utm8QReN9fyMS0dK40a5sVwsC7RRA,1970
 nebu/processors/processor.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nebu/redis/models.py,sha256=coPovAcVXnOU1Xh_fpJL4PO3QctgK9nBe5QYoqEcnxg,1230
 nebu/services/service.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nebu-0.1.6.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-nebu-0.1.6.dist-info/METADATA,sha256=1nMDXh8oVbrCvGVVkPCw26DJJZ26kZLBZu4odO8XGt8,1587
-nebu-0.1.6.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-nebu-0.1.6.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
-nebu-0.1.6.dist-info/RECORD,,
+nebu-0.1.8.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+nebu-0.1.8.dist-info/METADATA,sha256=MoqFzOItyQ4Knu3jfmjPEadooBFOPGZJTep7T0j1jeI,1587
+nebu-0.1.8.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+nebu-0.1.8.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
+nebu-0.1.8.dist-info/RECORD,,