nastasiasnk commited on
Commit
63a0234
1 Parent(s): b06d934

Update imports_utils.py

Browse files
Files changed (1) hide show
  1. imports_utils.py +5 -5
imports_utils.py CHANGED
@@ -194,14 +194,14 @@ def getDataFromSpeckle(
194
  dfLanduses = speckle_utils.get_dataframe(streamData, return_original_df=False)
195
  dfLanduses = dfLanduses.set_index("uuid", drop=False) # variable, uuid as default
196
 
197
- if type(landuseColumns) == type("s"):
198
- # extract landuse columns with "landuseColumns"
199
  landuse_columns = []
200
  for name in df_lu.columns:
201
- if name.startswith(landuseColumns):
202
- landuse_columns.append(name)
203
 
204
- elif type(landuseColumns) == type([]):
205
  #assmuming the user provided a lsit of columns
206
  landuse_columns = landuseColumns
207
 
 
194
  dfLanduses = speckle_utils.get_dataframe(streamData, return_original_df=False)
195
  dfLanduses = dfLanduses.set_index("uuid", drop=False) # variable, uuid as default
196
 
197
+ if type(landuseColumns) == type("s"):
198
+ # extract landuse columns with "landuseColumns"
199
  landuse_columns = []
200
  for name in df_lu.columns:
201
+ if name.startswith(landuseColumns):
202
+ landuse_columns.append(name)
203
 
204
+ elif type(landuseColumns) == type([]):
205
  #assmuming the user provided a lsit of columns
206
  landuse_columns = landuseColumns
207