#!/bin/bash
MA=/root/siton-data-2849d4ce327c4ccfb233ce33868fe7fe/zsy/old-road-code/envs/multimodal_affect/lib/python3.10/site-packages
DST=/opt/conda/envs/dlapo-py310-cu128/lib/python3.10/site-packages

for pkg in platformdirs sentry_sdk docker pynvml setproctitle; do
  if [ -d "$MA/$pkg" ]; then
    cp -r "$MA/$pkg" "$DST/" && echo "ok: $pkg"
  fi
  for dist in "$MA/${pkg}"-*.dist-info; do
    [ -d "$dist" ] && cp -r "$dist" "$DST/" && echo "ok: $(basename $dist)"
  done
done

echo "--- testing wandb import ---"
/opt/conda/envs/dlapo-py310-cu128/bin/python -c "import wandb; print('wandb ok')" 2>&1