From 370c30957201853d6f9104c97f0cd97dfb69e6e8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Arturo=20Filast=C3=B2?= Date: Wed, 4 Sep 2024 17:42:49 -0400 Subject: [PATCH] Fix call to make_observations_for_file_entry_batch --- .../temporal/activities/observations.py | 1 - oonipipeline/tests/test_ctrl.py | 16 +++++++--------- 2 files changed, 7 insertions(+), 10 deletions(-) diff --git a/oonipipeline/src/oonipipeline/temporal/activities/observations.py b/oonipipeline/src/oonipipeline/temporal/activities/observations.py index 4cf6dc1b..4149be05 100644 --- 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 index 949cf74c..a073397b 100644 --- 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