Add square step to arithmetic workflow. Add JSON write and read metho…

…ds to workflow model.
GeigerJ2 · May 5, 2025 · May 5, 2025 · May 5, 2025 · May 14, 2025 · May 5, 2025
commit 4f5f85d546343455de28a38d767e398f843a3d80
diff --git a/example_workflows/arithmetic/aiida.ipynb b/example_workflows/arithmetic/aiida.ipynb
diff --git a/example_workflows/arithmetic/workflow.py b/example_workflows/arithmetic/workflow.py
@@ -4,3 +4,6 @@ def get_prod_and_div(x, y):
 
 def get_sum(x, y):
     return x + y
+
+def square(x):
+    return x**2
diff --git a/python_workflow_definition/pyproject.toml b/python_workflow_definition/pyproject.toml
@@ -20,6 +20,8 @@ dependencies = [
   "numpy>=1.21,<2",
   "jobflow>=0.1.18,<=0.1.19",
   "pyiron_base>=0.11.10,<=0.11.11",
+  "pydantic",
+  "rich"
 ]
 
 [project.optional-dependencies]

diff --git a/python_workflow_definition/src/python_workflow_definition/_pydantic/__init__.py b/python_workflow_definition/src/python_workflow_definition/_pydantic/__init__.py
diff --git a/python_workflow_definition/src/python_workflow_definition/_pydantic/models.py b/python_workflow_definition/src/python_workflow_definition/_pydantic/models.py