mirror of
https://github.com/wassname/catalyst.git
synced 2026-06-30 14:20:48 +08:00
MIN: Changed isinstance check to allow more types.
This commit is contained in:
@@ -24,6 +24,7 @@ from copy import deepcopy
|
||||
from datetime import datetime
|
||||
from collections import deque
|
||||
from abc import ABCMeta, abstractmethod
|
||||
from numbers import Integral
|
||||
|
||||
import pandas as pd
|
||||
|
||||
@@ -394,7 +395,7 @@ class BatchTransform(EventWindow):
|
||||
self.compute_only_full = compute_only_full
|
||||
|
||||
self.sids = sids
|
||||
if isinstance(self.sids, (str, int)):
|
||||
if isinstance(self.sids, (basestring, Integral)):
|
||||
self.sids = [self.sids]
|
||||
|
||||
self.field_names = fields
|
||||
|
||||
Reference in New Issue
Block a user