From 7fd2308b85282d94fe541f6b55234d96e748c376 Mon Sep 17 00:00:00 2001
From: Matthew K Defenderfer <mdefende@uab.edu>
Date: Fri, 30 Aug 2024 09:04:35 -0500
Subject: [PATCH] acheck to see if mode exists in the dataset in the first
 place

---
 transfer-gpfs-with-s5cmd/fpart-db.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/transfer-gpfs-with-s5cmd/fpart-db.py b/transfer-gpfs-with-s5cmd/fpart-db.py
index 11ecc9e..26cf757 100755
--- a/transfer-gpfs-with-s5cmd/fpart-db.py
+++ b/transfer-gpfs-with-s5cmd/fpart-db.py
@@ -34,7 +34,8 @@ def main():
     ddf = dd.read_parquet(input_parquet)
     ddf = ddf.loc[ddf['path'].str.startswith(filter)].sort_values('path')
 
-    ddf = ddf.loc[~ddf['mode'].str.startswith('d')]
+    if 'mode' in ddf.columns:
+        ddf = ddf.loc[~ddf['mode'].str.startswith('d')]
 
     ddf['cmd'] = ddf['path'].map(lambda x: create_sync_cmd(x, filter=filter, dest=dest), meta=str)
     
-- 
GitLab