scratch/clear_cache.py

from afml.production.model_development import load_and_prepare_training_data

def clear_data_cache():
    print("Clearing cache for load_and_prepare_training_data...")
    try:
        load_and_prepare_training_data.cache_clear()
        print("Cache cleared successfully.")
    except Exception as e:
        print(f"Failed to clear cache: {e}")

if __name__ == "__main__":
    clear_data_cache()