Fix failing tests

Author: dyumanaditya

tests/unit/disable_jit/interpretations/test_interpretation_common.py

@@ -69,16 +69,12 @@ def add_edge(*args):
     ns.add_edge = add_edge
 
     _ground_rule_fn = _py(interpretation._ground_rule)
-    sig = inspect.signature(_ground_rule_fn)
-    has_num_ga = "num_ga" in sig.parameters
-    
-    def ground_rule(rule, interpretations_node, interpretations_edge, predicate_map_node, predicate_map_edge, nodes, edges, neighbors, reverse_neighbors, atom_trace, allow_ground_rules):
-        args = (rule, interpretations_node, interpretations_edge, predicate_map_node, predicate_map_edge, nodes, edges, neighbors, reverse_neighbors, atom_trace, allow_ground_rules)
-        if has_num_ga:
-            return _ground_rule_fn(*args, num_ga=[0], t=0, head_functions=())
-        else:
-            return _ground_rule_fn(*args, t=0, head_functions=())
-    
+    if "num_ga" in inspect.signature(_ground_rule_fn).parameters:
+        def ground_rule(*args, **kwargs):
+            return _ground_rule_fn(*args, num_ga=[0], **kwargs)
+    else:
+        def ground_rule(*args, **kwargs):
+            return _ground_rule_fn(*args, **kwargs)
     ns.ground_rule = ground_rule
     ns.update_rule_trace = _py(interpretation._update_rule_trace)
     ns.are_satisfied_node = _py(interpretation.are_satisfied_node)

tests/unit/disable_jit/interpretations/test_interpretation_init.py

@@ -76,8 +76,7 @@ def test_interpretation_init_neighbors(shim_types):
         False,
         False,
         False,
-        False,
-        "intersection",
+        0,
         False,
     )
     assert set(interp.neighbors["n1"]) == {"n2"}
@@ -212,8 +211,7 @@ def build_interp():
         False,
         False,
         False,
-        False,
-        "intersection",
+        0,
         False,
     )
     interp.time = 5