1
0
Fork 0

Update test_hill_climbing_search.py

master
lmoretti96 4 years ago committed by GitHub
parent 36f135b35a
commit 0088b449ce
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 4
      PyCTBN/tests/optimizers/test_hill_climbing_search.py

@ -54,9 +54,9 @@ class TestHillClimbingSearch(unittest.TestCase):
def test_structure_3(self): def test_structure_3(self):
with open("./PyCTBN/test_data/networks_and_trajectories_ternary_data_01_6_1.json") as f: with open("./PyCTBN/test_data/networks_and_trajectories_ternary_data_01_6_1.json") as f:
raw_data = json.load(f) raw_data = json.load(f)
trajectory_list_raw= raw_data["samples"] trajectory_list_raw= raw_data["samples"]
# Convert to DataFrame
trajectory_list = [pd.DataFrame(sample) for sample in trajectory_list_raw] trajectory_list = [pd.DataFrame(sample) for sample in trajectory_list_raw]
variables= raw_data["variables"] variables= raw_data["variables"]