Marthee commited on
Commit
768c885
·
verified ·
1 Parent(s): 96b07cc

Update tsadropboxretrieval.py

Browse files
Files changed (1) hide show
  1. tsadropboxretrieval.py +1 -1
tsadropboxretrieval.py CHANGED
@@ -248,7 +248,7 @@ def retrieveProjects(projname):
248
 
249
  # Filter RelevantDocuments and documentsToMeasure using the mask
250
  RelevantDocuments = parquetDf[mask][['name', 'path_display']].values.tolist()
251
- documentsToMeasure = RelevantDocuments # Filter documentsToMeasure for PDF files later if needed
252
 
253
  # Remove the temporary 'path_display_lower' column
254
  parquetDf.drop(columns=['path_display_lower'], inplace=True)
 
248
 
249
  # Filter RelevantDocuments and documentsToMeasure using the mask
250
  RelevantDocuments = parquetDf[mask][['name', 'path_display']].values.tolist()
251
+ documentsToMeasure = [doc for doc in RelevantDocuments if doc[0].endswith('.pdf')] # Filter documentsToMeasure for PDF files later if needed
252
 
253
  # Remove the temporary 'path_display_lower' column
254
  parquetDf.drop(columns=['path_display_lower'], inplace=True)