Mentions légales du service

Skip to content
Snippets Groups Projects
Verified Commit 16ec4236 authored by ANDREY Paul's avatar ANDREY Paul
Browse files

Silence mypy false-positives on pandas.

parent 3f6ed06d
No related branches found
No related tags found
No related merge requests found
Pipeline #793800 passed
...@@ -177,8 +177,8 @@ class InMemoryDataset(Dataset): ...@@ -177,8 +177,8 @@ class InMemoryDataset(Dataset):
return self.data return self.data
if isinstance(self.data, pd.DataFrame): if isinstance(self.data, pd.DataFrame):
if isinstance(self.f_cols[-1], str): if isinstance(self.f_cols[-1], str):
return self.data.loc[:, self.f_cols] return self.data.loc[:, self.f_cols] # type: ignore
return self.data.iloc[:, self.f_cols] return self.data.iloc[:, self.f_cols] # type: ignore
return self.data[:, self.f_cols] # type: ignore return self.data[:, self.f_cols] # type: ignore
@property @property
...@@ -187,7 +187,7 @@ class InMemoryDataset(Dataset): ...@@ -187,7 +187,7 @@ class InMemoryDataset(Dataset):
if (not self.expose_classes) or (self.target is None): if (not self.expose_classes) or (self.target is None):
return None return None
if isinstance(self.target, pd.DataFrame): if isinstance(self.target, pd.DataFrame):
return set(self.target.unstack().unique().tolist()) return set(self.target.unstack().unique().tolist()) # type: ignore
if isinstance(self.target, pd.Series): if isinstance(self.target, pd.Series):
return set(self.target.unique().tolist()) return set(self.target.unique().tolist())
if isinstance(self.target, np.ndarray): if isinstance(self.target, np.ndarray):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment