From d0f1fd687b78da5196451582713c9efe834d8b1c Mon Sep 17 00:00:00 2001 From: Thomas Wiecki Date: Tue, 10 Sep 2024 16:30:27 +0100 Subject: [PATCH] Fix pre-commit --- tests/mmm/test_lift_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/mmm/test_lift_test.py b/tests/mmm/test_lift_test.py index f827bc62..cb112042 100644 --- a/tests/mmm/test_lift_test.py +++ b/tests/mmm/test_lift_test.py @@ -14,10 +14,10 @@ import numpy as np import pandas as pd import pymc as pm -from pytensor.tensor.variable import TensorVariable import pytensor.tensor as pt import pytest from pymc.model_graph import fast_eval +from pytensor.tensor.variable import TensorVariable from sklearn.pipeline import Pipeline from sklearn.preprocessing import MaxAbsScaler @@ -35,8 +35,8 @@ create_time_varying_saturation, create_variable_indexer, exact_row_indices, - scale_lift_measurements, scale_channel_lift_measurements, + scale_lift_measurements, scale_target_for_lift_measurements, )