EnigmaConsultant's picture
Upload folder using huggingface_hub
deb302b verified
Raw
History Blame Contribute Delete
566 Bytes
import os, glob, joblib, sys
WORK = sys.argv[1]
# remove any stale markers
for m in glob.glob(os.path.join(WORK, "marker_*")):
os.remove(m)
for c in ["none","zlib","gzip","bz2","lzma","xz","lz4"]:
p = os.path.join(WORK, f"evil_{c}.joblib")
try:
joblib.load(p)
except Exception as ex:
print(f"{c:5s} load-exc {type(ex).__name__}: {ex}")
marker = os.path.join(WORK, f"marker_{c}")
if os.path.exists(marker):
print(f"{c:5s} EXECUTED=True {open(marker).read().strip()}")
else:
print(f"{c:5s} EXECUTED=False")