TypeError when loading dataset

[HenriqueCSJ]

I am receiving an error when trying to load load_organic_density (ChemML v 1.2)

from chemml.datasets import load_organic_density
smi, density, features = load_organic_density()

Here is the error:

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
Untitled-1.ipynb Cell 23 line 3
      [1](vscode-notebook-cell:Untitled-1.ipynb?jupyter-notebook#X31sdW50aXRsZWQ%3D?line=0) from chemml.datasets import load_organic_density
----> [3](vscode-notebook-cell:Untitled-1.ipynb?jupyter-notebook#X31sdW50aXRsZWQ%3D?line=2) smi, density, features = load_organic_density()

File [~\AppData\Roaming\Python\Python310\site-packages\chemml\datasets\base.py:87](https://untitled+.vscode-resource.vscode-cdn.net/~/AppData/Roaming/Python/Python310/site-packages/chemml/datasets/base.py:87), in load_organic_density()
     85 smi = pd.DataFrame(df['smiles'], columns=['smiles'])
     86 density = pd.DataFrame(df['density_Kg/m3'], columns=['density_Kg/m3'])
---> 87 features = df.drop(['smiles', 'density_Kg/m3'],1)
     89 return smi, density, features

TypeError: DataFrame.drop() takes from 1 to 2 positional arguments but 3 were given