Add test for creating a param-param edge

valohai · Jul 8, 2021 · 49a3de2 · 49a3de2
1 parent 8a5d270
commit 49a3de2
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/papi_tests/test_param_param.py b/papi_tests/test_param_param.py
@@ -0,0 +1,10 @@
+from papi import Papi
+from papi_examples.utils import read_config
+
+
+def test_param_param_edge():
+    papi = Papi(name="mypipeline", config=read_config("example1.yaml"))
+    t1 = papi.task("Train model")
+    t2 = papi.task("Train model")
+    t1.parameter_edge("learning_rate").to(t2.parameter_edge("learning_rate"))
+    assert len(papi.edges) == 1