[tool.poetry]
name = "assignment-1"
version = "0.1.0"
description = ""
authors = ["Your Name <you@example.com>"]
readme = "README.md"
packages = [{include = "assignment_1"}]

[tool.poetry.dependencies]
python = "3.10.4"
black = "*"
jupyterlab = "*"
ipython = "*"
numpy = "*"
pandas = "*"
jax = "*"
matplotlib = "*"
seaborn = "*"


torch = "*"
transformers = {extras = ["torch"], version = "*"}
sentencepiece = "*"
protobuf = "==3.20.*"


[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"