Update utilities/components.py

utilities/components.py

@@ -1,6 +1,5 @@
 import streamlit as st
 from utilities.template_helpers import upload_data
-from types import NoneType
 import extra_streamlit_components as stx
 
 import prince
@@ -19,7 +18,6 @@ import algos.regression.linR
 import algos.regression.ridge
 import algos.regression.elasticnet
 
-from types import NoneType
 
 def get_data(category, algo_name=None):
     if category in ['Classification','Regression']:
@@ -28,7 +26,7 @@ def get_data(category, algo_name=None):
         return train, test
     else:
         df = upload_data()
-        if type(df) != NoneType:
+        if df != None:
             return (df,)