Fix bug in saveDatasheet

syncrosim · Nov 2, 2023 · b1eeae4 · b1eeae4
1 parent df5555f
commit b1eeae4
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/R/saveDatasheet.R b/R/saveDatasheet.R
@@ -208,9 +208,12 @@ setMethod("saveDatasheet", signature(ssimObject = "SsimObject"), function(ssimOb
       }
     }
 
-    # Remove invalid column names\
-    cDat <- subset(cDat, select=-c(
-      ScenarioID,ProjectID,ScenarioName,ParentID,ParentName))
+    # Remove invalid column names
+    extraColNames <- c("ScenarioID","ProjectID","ScenarioName","ParentID","ParentName")
+    if (all(extraColNames %in% colnames(cDat))){
+      cDat <- subset(cDat, select=-c(
+        ScenarioID,ProjectID,ScenarioName,ParentID,ParentName)) 
+    }
 
     # convert factors to strings
     for (kk in seq(length.out = ncol(cDat))) {