Function find_cached_model_missing_shard

cake-core/src/utils/hf.rs:321–350

()

    #[test]
    fn find_cached_model_missing_shard() {
        let _lock = ENV_LOCK.lock().unwrap();
        let tmp = tempfile::tempdir().unwrap();
        let cache = tmp.path();
        std::env::set_var("HF_HUB_CACHE", cache.to_str().unwrap());

        let snap = cache
            .join("models--org--partial")
            .join("snapshots")
            .join("ghi789");
        fs::create_dir_all(&snap).unwrap();
        fs::write(snap.join("config.json"), "{}").unwrap();
        let index = serde_json::json!({
            "weight_map": {
                "layer.0.weight": "shard-00001.safetensors",
                "layer.1.weight": "shard-00002.safetensors"
            }
        });
        fs::write(
            snap.join("model.safetensors.index.json"),
            serde_json::to_string(&index).unwrap(),
        )
        .unwrap();
        fs::write(snap.join("shard-00001.safetensors"), "data").unwrap();

        let result = find_cached_model("org/partial");
        assert!(result.is_none());

        std::env::remove_var("HF_HUB_CACHE");
    }

    #[test]
    fn find_cached_model_not_present() {