Why are you defining the dataset here again? Just do from hepatitis import hepatitis_data