Skip to content

Commit

Permalink
Update predictor.py
Browse files Browse the repository at this point in the history
  • Loading branch information
RRobert92 committed Dec 28, 2024
1 parent dba1c9f commit 267bbc8
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions tardis_em/utils/predictor.py
Original file line number Diff line number Diff line change
Expand Up @@ -1116,8 +1116,10 @@ def get_file_list(self):
and not f.endswith(self.omit_format)
]

print(self.predict_list)
if self.dir.endswith((".mrc", ".rec", ".map", ".tif", ".tiff", ".am")):
self.dir = getcwd()
sys.exit()

# Update Dir paths
self.output = join(self.dir, "temp", "Predictions")
Expand Down

0 comments on commit 267bbc8

Please sign in to comment.