wip

substrate/core/models.py

@@ -66,7 +66,7 @@ class RunPythonIn(BaseModel):
 
 
 class RunPythonOut(BaseModel):
-    output: Optional[Dict[str, Any]] = None
+    output: Optional[Any] = None
     """
     Return value of your function.
     """

substrate/future_dataclass_models.py

@@ -110,7 +110,7 @@ class FutureRunPythonOut:
     (Future reference)
     Contents of stderr if your code did not run successfully.
     """
-    output: Optional[Dict[str, Any]] = None
+    output: Optional[Any] = None
     """
     (Future reference)
     Return value of your function.

substrate/run_python.py

@@ -11,14 +11,14 @@ class RunPython(CoreNode[RunPythonOut]):
     def __init__(
         self,
         function: Callable,
-        kwargs: Dict[str, Any],
+        kwargs: Dict[str, Any] = {},
         pip_install: Optional[List[str]] = None,
         hide: bool = False,
     ):
         """
         Args:
-            function: Function to run
-            kwargs: Keyword arguments to pass to your function
+            function: The function to run remotely.
+            kwargs: Keyword arguments to your function. This can include future references to the output of previous nodes.
             pip_install: List of pip dependencies to install in your function's sandbox
 
         https://guides.substrate.run/reference/run-python

substrate/typeddict_models.py

@@ -64,7 +64,7 @@ class RunPythonIn(TypedDict):
 
 
 class RunPythonOut(TypedDict):
-    output: NotRequired[Dict[str, Any]]
+    output: NotRequired[Any]
     """
     Return value of your function.
     """