fix: move to new definition, fix import

BIGmama-technology · Mar 9, 2024 · 0d199ad · 0d199ad
1 parent b6e765e
commit 0d199ad
Show file tree

Hide file tree

Showing 7 changed files with 18 additions and 10 deletions.
diff --git a/hyko_toolkit/functions/downloaders/youtube_downloader/main.py b/hyko_toolkit/functions/downloaders/youtube_downloader/main.py
@@ -9,7 +9,7 @@
 from pytube import YouTube
 
 from hyko_sdk.io import Video
-from hyko_sdk.types import Ext
+from hyko_sdk.models import Ext
 
 
 def download_video(url, resolution):

diff --git a/hyko_toolkit/functions/downloaders/youtube_downloader/metadata.py b/hyko_toolkit/functions/downloaders/youtube_downloader/metadata.py
@@ -2,7 +2,7 @@
 
 from pydantic import Field
 
-from hyko_sdk.definitions import SDKFunction
+from hyko_sdk.definitions import ToolkitFunction
 from hyko_sdk.io import Video
 from hyko_sdk.models import CoreModel
 
@@ -15,7 +15,11 @@ class Resolution(str, Enum):
     lowest = "lowest"
 
 
-func = SDKFunction(description="Download a video from YouTube.")
+func = ToolkitFunction(
+    name="youtube_downloader",
+    task="downloaders",
+    description="Download a video from YouTube.",
+)
 
 
 @func.set_input

diff --git a/hyko_toolkit/functions/plotting/bivariate_plot/main.py b/hyko_toolkit/functions/plotting/bivariate_plot/main.py
@@ -7,7 +7,7 @@
 from metadata import Inputs, Outputs, Params, func
 
 from hyko_sdk.io import Image
-from hyko_sdk.types import Ext
+from hyko_sdk.models import Ext
 
 
 def generate_violin_plot(x: list[float], y: list[float]):

diff --git a/hyko_toolkit/functions/plotting/bivariate_plot/metadata.py b/hyko_toolkit/functions/plotting/bivariate_plot/metadata.py
@@ -2,7 +2,7 @@
 
 from pydantic import Field
 
-from hyko_sdk.definitions import SDKFunction
+from hyko_sdk.definitions import ToolkitFunction
 from hyko_sdk.io import Image
 from hyko_sdk.models import CoreModel
 
@@ -18,7 +18,9 @@ class SupportedPlots(str, Enum):
     Scatter_Plot = "Scatter_Plot"
 
 
-func = SDKFunction(
+func = ToolkitFunction(
+    name="bivariate_plot",
+    task="plotting",
     description="Generate various types of plots with (X , Y) inputs.",
 )
 

diff --git a/hyko_toolkit/functions/plotting/univariate_plot/main.py b/hyko_toolkit/functions/plotting/univariate_plot/main.py
@@ -5,7 +5,7 @@
 from metadata import Inputs, Outputs, Params, func
 
 from hyko_sdk.io import Image
-from hyko_sdk.types import Ext
+from hyko_sdk.models import Ext
 
 
 def generate_pie_chart(y: list[float]):

diff --git a/hyko_toolkit/functions/plotting/univariate_plot/metadata.py b/hyko_toolkit/functions/plotting/univariate_plot/metadata.py
@@ -2,7 +2,7 @@
 
 from pydantic import Field
 
-from hyko_sdk.definitions import SDKFunction
+from hyko_sdk.definitions import ToolkitFunction
 from hyko_sdk.io import Image
 from hyko_sdk.models import CoreModel
 
@@ -12,7 +12,9 @@ class SupportedPlots(str, Enum):
     Histogram = "Histogram"
 
 
-func = SDKFunction(
+func = ToolkitFunction(
+    name="univariate_plot",
+    task="plotting",
     description="Generate various types of plots with Y input.",
 )
 

diff --git a/hyko_toolkit/models/computer_vision/background_image_removal/main.py b/hyko_toolkit/models/computer_vision/background_image_removal/main.py
@@ -2,7 +2,7 @@
 from rembg import remove
 
 from hyko_sdk.io import Image
-from hyko_sdk.types import Ext
+from hyko_sdk.models import Ext
 
 
 @func.on_execute