From b02a70b5a12114eeb933d0caf17d0b0c5027b08b Mon Sep 17 00:00:00 2001 From: adnaniazi Date: Thu, 11 Jul 2024 13:03:37 +0200 Subject: [PATCH] Reverted changes done to simplify install instructions --- pyproject.toml | 30 +++++++++++++++--------------- 1 file changed, 15 insertions(+), 15 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index ad3a88c..0824d5c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -59,27 +59,20 @@ pysam = "^0.22.1" typer = "^0.12.3" imbalanced-learn = "^0.12.3" -# Base JAX dependency -jax = {version = "^0.4.10", optional = true} - -# CPU installation -jax_cpu = {version = "^0.4.10", optional = true} - -# GPU installation (CUDA 12) -jax_gpu = {version = "^0.4.10", optional = true, source = "https://storage.googleapis.com/jax-releases/jax_cuda_releases.html"} - -# TPU installation -jax_tpu = {version = "^0.4.10", optional = true, source = "https://storage.googleapis.com/jax-releases/libtpu_releases.html"} +# JAX dependencies +jax = {version = "*", optional = true} [tool.poetry.extras] -cpu = ["jax_cpu"] -gpu = ["jax_gpu"] -tpu = ["jax_tpu"] +cpu = ["jax"] +gpu = ["jax"] +tpu = ["jax"] [[tool.poetry.source]] name = "pypi" priority = "primary" + + [tool.poetry.dev-dependencies] autoflake = "*" black = "*" @@ -116,7 +109,7 @@ profile = "black" src_paths = ["src", "tests"] [tool.black] -target-version = ["py37", "py38", "py39"] +target-version = ["py310", "py311", "py312"] include = '\.pyi?$' [tool.pytest.ini_options] @@ -148,3 +141,10 @@ ignore_missing_imports = true [tool.poetry.scripts] capfinder = "capfinder.cli:app" + + +#command1 = "package:module1:function1" +#command2 = "package:module2:function2" +#command3 = "package:module3:function3" + +[tool.setuptools_scm]