def cleanup_dir()

in 07_training/serverlessml/flowers/utils/util.py [0:0]

• 10 lines of code
• 3 McCabe index (conditional complexity)

def cleanup_dir(OUTPUT_DIR):
    on_cloud = OUTPUT_DIR.startswith("gs://")
    if on_cloud:
        try:
            subprocess.check_call("gsutil -m rm -r {}".format(OUTPUT_DIR).split())
        except subprocess.CalledProcessError:
            pass
    else:
        shutil.rmtree(OUTPUT_DIR, ignore_errors=True)
        os.makedirs(OUTPUT_DIR)