cavalab / fomo Goto Github PK

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
Cell In[2], line 1
----> 1 from fomo import FomoClassifier
      2 from pmlb import pmlb
      3 X,y = pmlb.fetch_data('adult', return_X_y=True)

File [..\lib\site-packages\fomo\__init__.py:1]
----> 1 from .estimator import FomoEstimator, FomoClassifier, FomoRegressor
      2 from .version import __version__

File [..\lib\site-packages\fomo\estimator.py:63]
     61 from pymoo.visualization.scatter import Scatter
     62 # types
---> 63 from types import NoneType
     64 from collections.abc import Callable
     67 class FomoEstimator(BaseEstimator):

ImportError: cannot import name 'NoneType' from 'types' ([..\lib\types.py]

from fomo import FomoClassifier
from pmlb import pmlb
X,y = pmlb.fetch_data('adult', return_X_y=True)
groups = ['race','sex']
est = FomoClassifier()
est.fit(X,y, protected_features=groups)