From fc24d46fa46efbd226b8fb1f7758abfbee8642a7 Mon Sep 17 00:00:00 2001 From: github-actions Date: Wed, 4 Oct 2023 12:09:13 +0000 Subject: [PATCH] style: linting --- .../test_flattened_dataset/test_add_values.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/timeseriesflattener/tests/test_timeseriesflattener/test_flattened_dataset/test_add_values.py b/src/timeseriesflattener/tests/test_timeseriesflattener/test_flattened_dataset/test_add_values.py index 68fae828..cde8ce10 100644 --- a/src/timeseriesflattener/tests/test_timeseriesflattener/test_flattened_dataset/test_add_values.py +++ b/src/timeseriesflattener/tests/test_timeseriesflattener/test_flattened_dataset/test_add_values.py @@ -1,6 +1,5 @@ """Tests for adding values to a flattened dataset.""" -import datetime as dt import numpy as np import pandas as pd @@ -500,6 +499,7 @@ def test_add_temporal_predictors_then_temporal_outcome(): check_dtype=False, ) + def test_add_temporal_incident_binary_outcome(): prediction_times_str = """entity_id,timestamp, 1,2021-11-05 00:00:00 @@ -571,7 +571,11 @@ def test_add_outcome_timestamps(): prediction_times_df = str_to_df(prediction_times_str) event_times_df = str_to_df(event_times_str) expected_df = str_to_df(expected_df_str) - expected_df['outc_timestamp_within_10_days_latest_fallback_nan_dichotomous'] = expected_df['outc_timestamp_within_10_days_latest_fallback_nan_dichotomous'].astype(str).replace('NaT', np.NaN) + expected_df["outc_timestamp_within_10_days_latest_fallback_nan_dichotomous"] = ( + expected_df["outc_timestamp_within_10_days_latest_fallback_nan_dichotomous"] + .astype(str) + .replace("NaT", np.NaN) + ) flattened_dataset = TimeseriesFlattener( prediction_times_df=prediction_times_df,