[FSTORE-1285] Model Dependent Transformation Functions (#206)

* load test * adding udf decorator to hopsworks repo * reverting pyproject modeified for loadtest * reverting pyproject modeified for loadtest * adding api documentation for udf decorator
logicalclocks · Jul 11, 2024 · e510018 · e510018
1 parent f21ac7e
commit e510018
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 1 deletion.
diff --git a/auto_doc.py b/auto_doc.py
@@ -27,6 +27,9 @@
         "mr_api": ["hopsworks.project.Project.get_model_registry"],
         "ms_api": ["hopsworks.project.Project.get_model_serving"],
     },
+    "api/udf.md": {
+        "udf": ["hopsworks.udf"],
+    },
     "api/connection.md": {
         "connection_create": ["hopsworks.connection.Connection.connection"],
         "connection_properties": keras_autodoc.get_properties(

diff --git a/docs/templates/api/udf.md b/docs/templates/api/udf.md
@@ -0,0 +1,3 @@
+# UDF API
+
+{{udf}}
diff --git a/mkdocs.yml b/mkdocs.yml
@@ -21,6 +21,7 @@ nav:
       - Login: generated/api/login.md
       - Connection: generated/api/connection.md
       - Projects: generated/api/projects.md
+      - UDF: generated/api/udf.md
       - Jobs: generated/api/jobs.md
       - Executions: generated/api/executions.md
       - FlinkCluster: generated/api/flink_cluster.md

diff --git a/python/hopsworks/__init__.py b/python/hopsworks/__init__.py
@@ -47,14 +47,16 @@
 _secrets_api = None
 _project_api = None
 
+udf = hsfs.hopsworks_udf.udf
+
 
 def hw_formatwarning(message, category, filename, lineno, line=None):
     return "{}: {}\n".format(category.__name__, message)
 
 
 warnings.formatwarning = hw_formatwarning
 
-__all__ = ["connection"]
+__all__ = ["connection", "udf"]
 
 logging.basicConfig(
     level=logging.INFO,