coverage

Author: rsomers1998

tests/surrogate_tests/test_causal_surrogate_assisted.py

@@ -56,7 +56,7 @@ def test_surrogate_model_generation(self):
 
         causal_dag = CausalDAG(self.dag_dot_path)
         z = Input("Z", int)
-        x = Input("X", int)
+        x = Input("X", float)
         m = Input("M", int)
         y = Output("Y", float)
         scenario = Scenario(variables={z, x, m, y})
@@ -75,7 +75,7 @@ def test_causal_surrogate_assisted_execution(self):
 
         causal_dag = CausalDAG(self.dag_dot_path)
         z = Input("Z", int)
-        x = Input("X", int)
+        x = Input("X", float)
         m = Input("M", int)
         y = Output("Y", float)
         scenario = Scenario(variables={z, x, m, y}, constraints={
@@ -107,7 +107,7 @@ def test_causal_surrogate_assisted_execution_failure(self):
 
         causal_dag = CausalDAG(self.dag_dot_path)
         z = Input("Z", int)
-        x = Input("X", int)
+        x = Input("X", float)
         m = Input("M", int)
         y = Output("Y", float)
         scenario = Scenario(variables={z, x, m, y}, constraints={
@@ -139,7 +139,7 @@ def test_causal_surrogate_assisted_execution_custom_aggregator(self):
 
         causal_dag = CausalDAG(self.dag_dot_path)
         z = Input("Z", int)
-        x = Input("X", int)
+        x = Input("X", float)
         m = Input("M", int)
         y = Output("Y", float)
         scenario = Scenario(variables={z, x, m, y}, constraints={
@@ -172,7 +172,7 @@ def test_causal_surrogate_assisted_execution_incorrect_search_config(self):
 
         causal_dag = CausalDAG(self.dag_dot_path)
         z = Input("Z", int)
-        x = Input("X", int)
+        x = Input("X", float)
         m = Input("M", int)
         y = Output("Y", float)
         scenario = Scenario(variables={z, x, m, y}, constraints={