[tool.poetry]
name = "Trajectron-plus-plus"
version = "0.1.1"
description = "This repository contains the code for Trajectron++: Dynamically-Feasible Trajectory Forecasting With Heterogeneous Data by Tim Salzmann*, Boris Ivanovic*, Punarjay Chakravarty, and Marco Pavone (* denotes equal contribution)."
authors = ["Ruben van de Ven <git@rubenvandeven.com>"]
readme = "README.md"
license = "MIT"

packages = [
    { include = "trajectron" },
]

[tool.poetry.scripts]
trajectron_train = "trajectron.train:main"

[tool.poetry.dependencies]
python = "^3.9,<3.12"
numpy = "^1.24.3"
opencv-python = "^4.7.0.72"
ipykernel = "^6.22.0"
torch = [
	{ version="1.12.1" },
#	{ url = "https://download.pytorch.org/whl/cu113/torch-1.12.1%2Bcu113-cp38-cp38-linux_x86_64.whl", markers = "python_version ~= '3.8' and sys_platform == 'linux'" },
	{ url = "https://download.pytorch.org/whl/cu113/torch-1.12.1%2Bcu113-cp310-cp310-linux_x86_64.whl", markers = "python_version ~= '3.10' and sys_platform == 'linux'" },
]
#torchvision = [
#	{ version="0.13.1" },
#	{ url = "https://download.pytorch.org/whl/cu113/torchvision-0.13.1%2Bcu113-cp38-cp38-linux_x86_64.whl", markers = "python_version ~= '3.8' and sys_platform == 'linux'" },
#	{ url = "https://download.pytorch.org/whl/cu113/torchvision-0.13.1%2Bcu113-cp310-cp310-linux_x86_64.whl", markers = "python_version ~= '3.10' and sys_platform == 'linux'" },
#]

#av = "^10.0.0"
matplotlib = "^3.5"
#numba = "^0.57.0"
#scikit-image = "^0.20.0"
#scikit-learn = "0.22.1"
#filterpy = "^1.4.5"
tqdm = "^4.65.0"
scipy = "^1.11.3"
pandas = "^2.1.1"
orjson = "^3.9.7"
nuscenes-devkit = "^1.1.11"
pyquaternion = "^0.9.9"
dill = "^0.3.7"
ncls = "^0.0.68"
notebook = "^7.0.4"
scikit-learn = "^1.3.1"
seaborn = "^0.13.0"
setuptools = "^68.2.2"
tensorboard = "^2.11"
tensorboardx = "^2.6.2.2"


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