requirements.txt

# --------------------------- #
#         Development         #
# --------------------------- #

# --------- data --------- #
botocore>=1.23.24,<1.23.25
awscli>=1.22.24,<1.22.25
boto3>=1.20.24,<1.20.25
s3fs>=2021.11.1
dvc==2.9.5

icecream
fire

# --------- loggers --------- #
# wandb
# neptune-client
mlflow
#mlflow>=1.1.0
# comet-ml
# tensorboard
torch-tb-profiler==0.4.0

# --------- linters --------- #
pre-commit      # hooks for applying linters on commit
click==8.0.4
black           # code formatting
isort           # import sorting
flake8          # code analysis
nbstripout      # remove output from jupyter notebooks
pre-commit

# --------- others --------- #
python-dotenv   # loading env variables from .env file
rich            # beautiful text formatting in terminal
sh              # for running bash commands in some tests
pytest          # tests
pudb            # debugger


# --------------------------- #
#           project           #
# --------------------------- #

# --------- pytorch --------- #
torch>=1.10.0
torchvision>=0.11.0
pytorch-lightning==1.6.4
torchmetrics>=0.7.0
torchtext
transformers
timm

# --------- hydra --------- #
hydra-core>=1.1.0
hydra-colorlog>=1.1.0
hydra-optuna-sweeper<1.2.0
omegaconf

# ---- computer-vision ---- #
opencv-python>=4.5.5.64

# --------- basics --------- #
numpy
nptyping
transformers
pandas
Pillow==9.0.1

# --------- inference --------- #
# fastapi
# uvicorn

# --------- others --------- #
tqdm
protobuf<3.20
av
gradio==2.7.5