Update main.py

main.py

@@ -37,31 +37,21 @@ storage_options={'account_name': AZURE_STORAGE_ACCOUNT, 'account_key': AZURE_STO
 
 df = pl.read_parquet(filepath,storage_options=storage_options)
 
-#pq.write_table(polars_dataframe.to_arrow(), 'liu/file_name.parquet', filesystem=abfs)
-#storage_options = {
-#    "aws_access_key_id": ,
-#    "aws_secret_access_key": os.environ['AZURE_STORAGE_ACCESS_KEY'],
-#}
-#df = pl.scan_parquet(source, storage_options=storage_options)
-
-
-#config_path = "./azure/data/config.yaml"
+# Load in config file
+config_path = "./data/config.yaml"
 
 # Add the read-in data from the yaml file
-#def read_config(filename):
-#    with open(filename, 'r') as yaml_file:
-#        config = yaml.safe_load(yaml_file)
-#   return config
+def read_config(filename):
+    with open(filename, 'r') as yaml_file:
+        config = yaml.safe_load(yaml_file)
+    return config
 
-#config = read_config(config_path)
+config = read_config(config_path)
 #path_parquet = config.get("path_parquet")
-#conditions = config.get("conditions")
-#col_features = config.get("col_features")
-#col_counts = config.get("col_counts")
-#col_mt = config.get("col_mt")
-
-# Import the data from one .parquet file
-#df = pl.read_parquet(path_parquet)
+conditions = config.get("conditions")
+col_features = config.get("col_features")
+col_counts = config.get("col_counts")
+col_mt = config.get("col_mt")
 #df = df.rename({"__index_level_0__": "Unnamed: 0"})
 
 # Setup the app