cheminee-similarity-model 0.1.3

Builds a Morgan fingerprint encoder model to accelerate similarity searches
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

use flate2::bufread::GzDecoder;
use std::fs::File;
use std::io::BufReader;
use tar::Archive;

fn main() {
    let asset_bytes = reqwest::blocking::get("https://cheminee-models.s3.eu-central-1.amazonaws.com/similarity/similarity-0.1.0.tar.gz")
        .expect("Failed get request")
        .bytes()
        .expect("Failed to retrieve bytes");

    let out_dir = std::env::var("OUT_DIR").expect("OUT_DIR is not set");
    let tar_gz_path = format!("{}/similarity-0.1.0.tar.gz", &out_dir);

    std::fs::write(&tar_gz_path, asset_bytes).expect("Failed to write tar file");

    let tar_gz_file = File::open(&tar_gz_path).expect("Failed to open tar file");
    let decoder = GzDecoder::new(BufReader::new(tar_gz_file));

    let mut archive = Archive::new(decoder);
    archive.unpack(out_dir).expect("Failed to unpack tar ball");
}