rename

SubstrateLabs · Aug 6, 2024 · 8b640e0 · 8b640e0
1 parent 99928cd
commit 8b640e0
Show file tree

Hide file tree

Showing 5 changed files with 19 additions and 19 deletions.
diff --git a/substrate/__init__.py b/substrate/__init__.py
@@ -29,6 +29,7 @@
     SegmentAnything,
     BatchComputeJSON,
     BatchComputeText,
+    InterpolateFrame,
     ListVectorStores,
     Llama3Instruct8B,
     MultiComputeJSON,
@@ -41,7 +42,6 @@
     Mistral7BInstruct,
     MultiInpaintImage,
     SegmentUnderPoint,
-    FrameInterpolation,
     MultiGenerateImage,
     Mixtral8x7BInstruct,
     StableVideoDiffusion,
@@ -84,7 +84,7 @@
     "StableDiffusionXLInpaint",
     "StableDiffusionXLControlNet",
     "StableVideoDiffusion",
-    "FrameInterpolation",
+    "InterpolateFrame",
     "TranscribeSpeech",
     "GenerateSpeech",
     "RemoveBackground",

diff --git a/substrate/core/models.py b/substrate/core/models.py
@@ -1003,7 +1003,7 @@ class Config:
     """
 
 
-class FrameInterpolationIn(BaseModel):
+class InterpolateFrameIn(BaseModel):
     class Config:
         extra = Extra.allow
 
@@ -1029,7 +1029,7 @@ class Config:
     """
 
 
-class FrameInterpolationOut(BaseModel):
+class InterpolateFrameOut(BaseModel):
     class Config:
         extra = Extra.allow
 

diff --git a/substrate/future_dataclass_models.py b/substrate/future_dataclass_models.py
@@ -1272,9 +1272,9 @@ class FutureStableVideoDiffusionOut:
 
 
 @dataclass
-class FutureFrameInterpolationIn:
+class FutureInterpolateFrameIn:
     """
-    Future reference to FutureFrameInterpolationIn
+    Future reference to FutureInterpolateFrameIn
     """
 
     frame_uris: List[str]
@@ -1305,9 +1305,9 @@ class FutureFrameInterpolationIn:
 
 
 @dataclass
-class FutureFrameInterpolationOut:
+class FutureInterpolateFrameOut:
     """
-    Future reference to FutureFrameInterpolationOut
+    Future reference to FutureInterpolateFrameOut
     """
 
     video_uri: Optional[str] = None

diff --git a/substrate/nodes.py b/substrate/nodes.py
@@ -38,6 +38,7 @@
         SegmentAnythingOut,
         BatchComputeJSONOut,
         BatchComputeTextOut,
+        InterpolateFrameOut,
         ListVectorStoresOut,
         Llama3Instruct8BOut,
         MultiComputeJSONOut,
@@ -50,7 +51,6 @@
         Mistral7BInstructOut,
         MultiInpaintImageOut,
         SegmentUnderPointOut,
-        FrameInterpolationOut,
         MultiGenerateImageOut,
         Mixtral8x7BInstructOut,
         StableVideoDiffusionOut,
@@ -89,6 +89,7 @@
     FutureSegmentAnythingOut,
     FutureBatchComputeJSONOut,
     FutureBatchComputeTextOut,
+    FutureInterpolateFrameOut,
     FutureListVectorStoresOut,
     FutureLlama3Instruct8BOut,
     FutureMultiComputeJSONOut,
@@ -101,7 +102,6 @@
     FutureMistral7BInstructOut,
     FutureMultiInpaintImageOut,
     FutureSegmentUnderPointOut,
-    FutureFrameInterpolationOut,
     FutureMultiGenerateImageOut,
     FutureMixtral8x7BInstructOut,
     FutureStableVideoDiffusionOut,
@@ -999,8 +999,8 @@ def future(self) -> FutureStableVideoDiffusionOut:  # type: ignore
         return super().future  # type: ignore
 
 
-class FrameInterpolation(CoreNode[FrameInterpolationOut]):
-    """https://substrate.run/nodes#FrameInterpolation"""
+class InterpolateFrame(CoreNode[InterpolateFrameOut]):
+    """https://substrate.run/nodes#InterpolateFrame"""
 
     def __init__(
         self,
@@ -1020,7 +1020,7 @@ def __init__(
             fps: Frames per second of the generated video.
             num_rounds: Number of rounds of interpolation. Each round interpolates between all adjacent frames. This also includes the interpolated frames from the previous round.
 
-        https://substrate.run/nodes#FrameInterpolation
+        https://substrate.run/nodes#InterpolateFrame
         """
         super().__init__(
             frame_uris=frame_uris,
@@ -1029,17 +1029,17 @@ def __init__(
             fps=fps,
             num_rounds=num_rounds,
             hide=hide,
-            out_type=FrameInterpolationOut,
+            out_type=InterpolateFrameOut,
             **kwargs,
         )
-        self.node = "FrameInterpolation"
+        self.node = "InterpolateFrame"
 
     @property
-    def future(self) -> FutureFrameInterpolationOut:  # type: ignore
+    def future(self) -> FutureInterpolateFrameOut:  # type: ignore
         """
         Future reference to this node's output.
 
-        https://substrate.run/nodes#FrameInterpolation
+        https://substrate.run/nodes#InterpolateFrame
         """
         return super().future  # type: ignore
 

diff --git a/substrate/typeddict_models.py b/substrate/typeddict_models.py
@@ -855,7 +855,7 @@ class StableVideoDiffusionOut(TypedDict):
     """
 
 
-class FrameInterpolationIn(TypedDict):
+class InterpolateFrameIn(TypedDict):
     frame_uris: NotRequired[List[str]]
     """
     Frames.
@@ -878,7 +878,7 @@ class FrameInterpolationIn(TypedDict):
     """
 
 
-class FrameInterpolationOut(TypedDict):
+class InterpolateFrameOut(TypedDict):
     video_uri: NotRequired[str]
     """
     Generated video.