UnboundLocalError: local variable 'array' referenced before assignment

[qiriro]

@BorjaRei @manwestc , great library you've created! It's so helpful, and I have recently been exploring how to create something similar myself. I'm glad you created it before I did.

I found a small bug that needs fixing. In the model.py the following code generates UnboundLocalError: local variable 'array' referenced before assignment

# Read the CSV
       if type(data)==str:
           dataset = pd.read_csv(data)
           array = dataset.values
       elif type(data)==pd:
           array = data.values

The elif statement in this code is incorrect. Here's a corrected version:

 # Read the CSV
        if type(data)==str:
            dataset = pd.read_csv(data)
            array = dataset.values
        elif isinstance(data, pd.DataFrame):
            array = data.values

[BorjaRei]

You are totally right. This isn't the final version, we will be fixing bugs and adding functionalities in the next weeks.
Thank you!