fxn 0.0.46__tar.gz → 0.0.48__tar.gz

{fxn-0.0.46 → fxn-0.0.48}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fxn
-Version: 0.0.46
+Version: 0.0.48
 Summary: Run prediction functions locally in Python. Register at https://fxn.ai.
 Author-email: "NatML Inc." <hi@fxn.ai>
 License:                                  Apache License

{fxn-0.0.46 → fxn-0.0.48}/fxn/beta/metadata.py

@@ -27,7 +27,7 @@ def _validate_ort_inference_session (session: "onnxruntime.InferenceSession") ->
 
 class CoreMLInferenceMetadata (BaseModel):
     """
-    Metadata required to lower PyTorch models for inference on iOS, macOS, and visionOS with CoreML.
+    Metadata required to lower a PyTorch model for inference on iOS, macOS, and visionOS with CoreML.
     """
     kind: Literal["meta.inference.coreml"] = "meta.inference.coreml"
     model: Annotated[object, BeforeValidator(_validate_torch_module)] = Field(description="PyTorch module to apply metadata to.")
@@ -36,7 +36,7 @@ class CoreMLInferenceMetadata (BaseModel):
 
 class ONNXInferenceMetadata (BaseModel):
     """
-    Metadata required to lower PyTorch models for inference.
+    Metadata required to lower a PyTorch model for inference.
     """
     kind: Literal["meta.inference.onnx"] = "meta.inference.onnx"
     model: Annotated[object, BeforeValidator(_validate_torch_module)] = Field(description="PyTorch module to apply metadata to.")
@@ -45,16 +45,25 @@ class ONNXInferenceMetadata (BaseModel):
 
 class ONNXRuntimeInferenceSessionMetadata (BaseModel):
     """
-    Metadata required to lower ONNXRuntime inference sessions for inference.
+    Metadata required to lower an ONNXRuntime `InferenceSession` for inference.
     """
     kind: Literal["meta.inference.onnxruntime"] = "meta.inference.onnxruntime"
     session: Annotated[object, BeforeValidator(_validate_ort_inference_session)] = Field(description="ONNXRuntime inference session to apply metadata to.")
     model_path: Path = Field(description="ONNX model path. The model must exist at this path in the compiler sandbox.")
     model_config = ConfigDict(arbitrary_types_allowed=True, frozen=True)
 
-class GGUFInferenceMetadata (BaseModel): # INCOMPLETE
+class LiteRTInferenceMetadata (BaseModel):
     """
-    Metadata required to lower GGUF models for LLM inference.
+    Metadata required to lower PyTorch model for inference with LiteRT (fka TensorFlow Lite).
+    """
+    kind: Literal["meta.inference.litert"] = "meta.inference.litert"
+    model: Annotated[object, BeforeValidator(_validate_torch_module)] = Field(description="PyTorch module to apply metadata to.")
+    model_args: list[object] = Field(description="Positional inputs to the model.")
+    model_config = ConfigDict(arbitrary_types_allowed=True, frozen=True)
+
+class LlamaCppInferenceMetadata (BaseModel): # INCOMPLETE
+    """
+    Metadata required to lower a GGUF model for LLM inference.
     """
     kind: Literal["meta.inference.gguf"] = "meta.inference.gguf"
     model_path: Path = Field(description="GGUF model path. The model must exist at this path in the compiler sandbox.")

{fxn-0.0.46 → fxn-0.0.48}/fxn/sandbox.py

@@ -43,7 +43,8 @@ class UploadDirectoryCommand (UploadableCommand):
 
     def get_files (self) -> list[Path]:
         from_path = Path(self.from_path)
-        assert from_path.is_absolute(), "Cannot upload directory because directory path must be absolute"
+        if not from_path.is_absolute():
+            raise ValueError("Cannot upload directory because directory path must be absolute")
         return [file for file in from_path.rglob("*") if file.is_file()]
     
 class EntrypointCommand (UploadableCommand):
@@ -106,7 +107,11 @@ class Sandbox (BaseModel):
             from_path (str | Path): File path on the local file system.
             to_path (str | Path): Remote path to upload file to.
         """
-        command = UploadFileCommand(from_path=str(from_path), to_path=str(to_path))
+        from_path = from_path if isinstance(from_path, Path) else Path(from_path)
+        command = UploadFileCommand(
+            from_path=str(from_path.resolve()),
+            to_path=str(to_path)
+        )
         return Sandbox(commands=self.commands + [command])
 
     def upload_directory (
@@ -121,7 +126,11 @@ class Sandbox (BaseModel):
             from_path (str | Path): Directory path on the local file system.
             to_path (str | Path): Remote path to upload directory to.
         """
-        command = UploadDirectoryCommand(from_path=str(from_path), to_path=str(to_path))
+        from_path = from_path if isinstance(from_path, Path) else Path(from_path)
+        command = UploadDirectoryCommand(
+            from_path=str(from_path.resolve()),
+            to_path=str(to_path)
+        )
         return Sandbox(commands=self.commands + [command])
 
     def pip_install (self, *packages: str) -> Sandbox:
@@ -180,7 +189,8 @@ class Sandbox (BaseModel):
         return self
 
     def __upload_file (self, path: Path, fxn: Function) -> str:
-        assert path.is_file(), "Cannot upload file at path {path} because it is not a file"
+        if not path.is_file():
+            raise ValueError(f"Cannot upload file at path {path} because it is not a file")
         hash = self.__compute_hash(path)
         try:
             fxn.client.request(method="HEAD", path=f"/resources/{hash}")

{fxn-0.0.46 → fxn-0.0.48}/fxn/version.py

@@ -3,4 +3,4 @@
 #   Copyright © 2025 NatML Inc. All Rights Reserved.
 #
 
-__version__ = "0.0.46"
+__version__ = "0.0.48"

{fxn-0.0.46 → fxn-0.0.48}/fxn.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fxn
-Version: 0.0.46
+Version: 0.0.48
 Summary: Run prediction functions locally in Python. Register at https://fxn.ai.
 Author-email: "NatML Inc." <hi@fxn.ai>
 License:                                  Apache License