Fix call to make_observations_for_file_entry_batch

ooni · Sep 4, 2024 · 370c309 · 370c309
1 parent 3481d4a
commit 370c309
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 10 deletions.
diff --git a/oonipipeline/src/oonipipeline/temporal/activities/observations.py b/oonipipeline/src/oonipipeline/temporal/activities/observations.py
@@ -113,7 +113,6 @@ def make_observations_for_file_entry_batch(
                             continue
                         msmt = None
                         try:
-                            t = PerfTimer()
                             msmt = load_measurement(msmt_dict)
                             if not msmt.test_keys:
                                 log.error(

diff --git a/oonipipeline/tests/test_ctrl.py b/oonipipeline/tests/test_ctrl.py
@@ -60,15 +60,13 @@ def test_web_ground_truth_from_clickhouse(db, datadir, netinfodb, tmp_path):
         )
     ]
     obs_msmt_count = make_observations_for_file_entry_batch(
-        MakeObservationsFileEntryBatch(
-            file_entry_batch=file_entry_batch,
-            clickhouse=db.clickhouse_url,
-            write_batch_size=1,
-            data_dir=datadir,
-            bucket_date="2023-10-31",
-            probe_cc=["US"],
-            fast_fail=False,
-        )
+        file_entry_batch=file_entry_batch,
+        clickhouse=db.clickhouse_url,
+        write_batch_size=1,
+        data_dir=datadir,
+        bucket_date="2023-10-31",
+        probe_cc=["US"],
+        fast_fail=False,
     )
     assert obs_msmt_count == 299
     # Wait for buffers to flush