Merge branch 'feature/env_config' of https://github.com/VectorInstitute/vector-inference into feature/env_config

Author: XkunW

docs/index.md

@@ -16,4 +16,3 @@ If you'd like to use `vec-inf` on your own Slurm cluster, you would need to upda
 * Clone the repository and update the `environment.yaml` and the `models.yaml` file in [`vec_inf/config`](vec_inf/config/), then install from source by running `pip install .`.
 * The package would try to look for cached configuration files in your environment before using the default configuration. The default cached configuration directory path points to `/model-weights/vec-inf-shared`, you would need to create an `environment.yaml` and a `models.yaml` following the format of these files in [`vec_inf/config`](vec_inf/config/).
 * The package would also look for an enviroment variable `VEC_INF_CONFIG_DIR`. You can put your `environment.yaml` and `models.yaml` in a directory of your choice and set the enviroment variable `VEC_INF_CONFIG_DIR` to point to that location.
-

vec_inf/client/_slurm_vars.py

@@ -3,7 +3,7 @@
 import os
 import warnings
 from pathlib import Path
-from typing import Any, TypeAlias
+from typing import Any
 
 import yaml
 from typing_extensions import Literal

vec_inf/config/environment.yaml

@@ -24,4 +24,4 @@ default_args:
   partition: "a40"
   data_type: "auto"
   log_dir: "~/.vec-inf-logs"
-  model_weights_parent_dir: "/model-weights"
+  model_weights_parent_dir: "/model-weights"