Spaces:
Sleeping
Sleeping
Update imports_utils.py
Browse files- 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 |
-
|
198 |
-
|
199 |
landuse_columns = []
|
200 |
for name in df_lu.columns:
|
201 |
-
|
202 |
-
|
203 |
|
204 |
-
|
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 |
|