Skip to content

Commit

Permalink
lint fix
Browse files Browse the repository at this point in the history
  • Loading branch information
Tamar Grey committed Feb 15, 2024
1 parent f9b78dd commit df663d3
Show file tree
Hide file tree
Showing 5 changed files with 10 additions and 8 deletions.
9 changes: 6 additions & 3 deletions featuretools/entityset/entityset.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
import copy
import logging
from pdb import set_trace
import warnings
from collections import defaultdict

Expand Down Expand Up @@ -1275,9 +1274,13 @@ def add_last_time_indexes(self, updated_dataframes=None):
if lti_is_spark:
# TODO: Figure out a workaround for fillna and replace
if lti_df["last_time_old"].dtype == "float64":
lti_df["last_time_old"] = ps.to_datetime(lti_df["last_time_old"])
lti_df["last_time_old"] = ps.to_datetime(
lti_df["last_time_old"],
)
if lti_df["last_time"].dtype == "float64":
lti_df["last_time"] = ps.to_datetime(lti_df["last_time"])
lti_df["last_time"] = ps.to_datetime(
lti_df["last_time"],
)
lti_df = lti_df.max(axis=1)
else:
lti_df["last_time"] = lti_df["last_time"].astype(
Expand Down
1 change: 0 additions & 1 deletion featuretools/tests/entityset_tests/test_ww_es.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
from datetime import datetime
from pdb import set_trace

import numpy as np
import pandas as pd
Expand Down
Original file line number Diff line number Diff line change
@@ -1,9 +1,7 @@
from inspect import isclass
from pdb import set_trace

import numpy as np
import pandas as pd
import pyarrow as pa
import pytest
from woodwork.column_schema import ColumnSchema
from woodwork.logical_types import (
Expand Down Expand Up @@ -353,6 +351,7 @@ def test_not_equal_categorical(simple_es):
False,
]


def test_not_equal_different_dtypes(simple_es):
f1 = Feature(
[
Expand Down
3 changes: 2 additions & 1 deletion featuretools/tests/testing_utils/generate_fake_dataframe.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
import random
import pytest
from datetime import datetime as dt

import pandas as pd
import pytest
import woodwork.type_sys.type_system as ww_type_system
from woodwork import logical_types

Expand Down Expand Up @@ -43,6 +43,7 @@ def generate_fake_dataframe(
):
dask = pytest.importorskip("dask", reason="Dask not installed, skipping")
dask.config.set({"dataframe.convert-string": False})

def randomize(values_):
random.seed(10)
values = values_.copy()
Expand Down
2 changes: 1 addition & 1 deletion featuretools/tests/testing_utils/mock_ds.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
import pytest
from datetime import datetime

import numpy as np
import pandas as pd
import pytest
from woodwork.logical_types import (
URL,
Boolean,
Expand Down

0 comments on commit df663d3

Please sign in to comment.