#!/usr/bin/env bash
mkdir -p stock_transformer_back
cd stock_transformer_back


STUDIO="hallucination/slow-rose-kq8y"

########################
# 1. 目录：先 mkdir，再下载
########################

# configs/
mkdir -p configs
lightning download folder stock_transformer_back/configs \
  --studio "$STUDIO" \
  --local-path ./configs

# data/
mkdir -p data
lightning download folder stock_transformer_back/data \
  --studio "$STUDIO" \
  --local-path ./data

# data_provider/
mkdir -p data_provider
lightning download folder stock_transformer_back/data_provider \
  --studio "$STUDIO" \
  --local-path ./data_provider

# exp/
mkdir -p exp
lightning download folder stock_transformer_back/exp \
  --studio "$STUDIO" \
  --local-path ./exp

# layers/
mkdir -p layers
lightning download folder stock_transformer_back/layers \
  --studio "$STUDIO" \
  --local-path ./layers

# models/
mkdir -p models
lightning download folder stock_transformer_back/models \
  --studio "$STUDIO" \
  --local-path ./models

# old_stuff/
mkdir -p old_stuff
lightning download folder stock_transformer_back/old_stuff \
  --studio "$STUDIO" \
  --local-path ./old_stuff

# utils/
mkdir -p utils
lightning download folder stock_transformer_back/utils \
  --studio "$STUDIO" \
  --local-path ./utils

# 注意：不下 lightning_logs，所以这里不写它


########################
# 2. 单个文件：直接下到当前目录
########################

lightning download file stock_transformer_back/.gitignore     --studio "$STUDIO"
lightning download file stock_transformer_back/.tmux.conf     --studio "$STUDIO"
lightning download file stock_transformer_back/LICENSE        --studio "$STUDIO"
lightning download file stock_transformer_back/README.md      --studio "$STUDIO"
lightning download file stock_transformer_back/Stockformer.py --studio "$STUDIO"
lightning download file stock_transformer_back/clean_ipynb.sh --studio "$STUDIO"

lightning download file stock_transformer_back/data_collect.ipynb  --studio "$STUDIO"
lightning download file stock_transformer_back/data_collect.py     --studio "$STUDIO"
lightning download file stock_transformer_back/data_loader.py      --studio "$STUDIO"
lightning download file stock_transformer_back/data_prepare.ipynb  --studio "$STUDIO"
lightning download file stock_transformer_back/data_prepare.py     --studio "$STUDIO"

lightning download file stock_transformer_back/embed.py        --studio "$STUDIO"
lightning download file stock_transformer_back/exp_timeseries.py --studio "$STUDIO"

lightning download file stock_transformer_back/run_bbtest.py   --studio "$STUDIO"
lightning download file stock_transformer_back/run_hypopt.py   --studio "$STUDIO"
lightning download file stock_transformer_back/run_once.py     --studio "$STUDIO"

lightning download file stock_transformer_back/stock_metrics.py --studio "$STUDIO"
lightning download file stock_transformer_back/tools.py         --studio "$STUDIO"

lightning download file stock_transformer_back/vis_data.ipynb    --studio "$STUDIO"
lightning download file stock_transformer_back/vis_results.ipynb --studio "$STUDIO"

lightning download file stock_transformer_back/wavelet.py        --studio "$STUDIO"