msc-patryk-bartkowiak/code/pyproject.toml

[project]
name = "codebert-training"
version = "1.0.0"
description = "This code is used to train CodeBERT"
authors = [
    {name = "Patryk Bartkowiak", email = "patbar15@st.amu.edu.pl"},
]
dependencies = [
    "wandb==0.18.5",
    "torch==2.5.0",
    "tqdm==4.66.5",
    "tree-sitter==0.23.1",
    "transformers==4.45.2",
    "datasets==3.0.1",
    "huggingface-hub==0.26.0",
    "matplotlib==3.9.2",
    "scikit-learn==1.5.2",
    "seaborn==0.13.2",
]
requires-python = "==3.11.*"
readme = "README.md"
license = {text = "MIT"}

[[tool.pdm.source]]
url = "https://download.pytorch.org/whl/cu124"
verify_ssl = true
include_packages = ["torch", "torchvision", "torchaudio"]
name = "cuda"

[build-system]
requires = ["pdm-backend"]
build-backend = "pdm.backend"

[tool.pdm]
distribution = true

[tool.pdm.scripts]
run_training = {cmd = "src/train_codebert_mlm.py"}
run_tree_training = {cmd = "src/train_tree_codebert_mlm.py"}
parse_dataset = {cmd = "src/parse_dataset.py"}