Skip to content

Commit

Permalink
Update pyproject.toml dependencies and env_*.yml files
Browse files Browse the repository at this point in the history
Use dev version of acnportal
  • Loading branch information
chrisyeh96 committed Nov 17, 2023
1 parent 3b65db1 commit d50bdc4
Show file tree
Hide file tree
Showing 3 changed files with 28 additions and 16 deletions.
8 changes: 2 additions & 6 deletions env_cogen.yml
Original file line number Diff line number Diff line change
Expand Up @@ -13,11 +13,8 @@
# - only supports gymnasium 0.28.1 (see https://github.com/ray-project/ray/blob/ray-2.8.0/python/setup.py#L305)
# - officially seems to only supports pettingzoo 1.23.1 (see https://github.com/ray-project/ray/blob/ray-2.8.0/python/requirements/ml/rllib-test-requirements.txt),
# but empirically seems to work with pettingzoo 1.24.*
# - Onnxruntime 1.16:
# - pre-built pip package only supports up to CUDA 11.8 (see https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html)
# - PyTorch 2.1 supports both CUDA 11.8 and 12.1
#
# last updated: November 13, 2023
# last updated: November 16, 2023
name: sustaingym_cogen
channels:
- pytorch # for pytorch
Expand Down Expand Up @@ -47,11 +44,10 @@ dependencies:
- gymnasium==0.28.1
- pettingzoo==1.24.1
- ray[rllib]==2.8.*
- onnxruntime==1.16.* # the ONNX model for CogenEnv is small and runs sufficiently fast on CPU

# uncomment for CPU-only
# - onnxruntime==1.16.*
# - tensorflow==2.14.*

# for GPU. comment out for CPU-only.
- onnxruntime-gpu==1.16.*
- tensorflow[and-cuda]==2.14.*
3 changes: 1 addition & 2 deletions env_ev.yml
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@
# - latest official release v0.3.2 only supports up to Pandas 1.1,
# and Pandas 1.1 only supports up to Python 3.9
# - we install from the ACNPortal dev branch, which includes a fix that supports Pandas 2
#
# - TensorFlow 2.14:
# - the GPU version only works with Python <=3.10 (see https://github.com/tensorflow/tensorflow/issues/61986)
# - TensorFlow 2.15 should fix this issue
Expand All @@ -19,7 +18,7 @@
# - officially seems to only supports pettingzoo 1.23.1 (see https://github.com/ray-project/ray/blob/ray-2.8.0/python/requirements/ml/rllib-test-requirements.txt),
# but empirically seems to work with pettingzoo 1.24.*
#
# last updated: November 13, 2023
# last updated: November 16, 2023
name: sustaingym_ev
channels:
- pytorch # for pytorch
Expand Down
33 changes: 25 additions & 8 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -24,32 +24,49 @@ dependencies = [

[project.optional-dependencies]
# Update dependencies in `all` if any are added or removed
evcharging = [
"acnportal==0.3.2",
all = [
"acnportal@git+https://github.com/zach401/acnportal@0127a055cf",
"cvxpy>=1.3, <2", # semver
"mosek>=10.0.44, <11", # semver
"pandas==1.1.*", # semver
"openpyxl>=3.1",
"onnxruntime>=1.15, <2", # semver
"pandas>=1.1, <3", # semver
"pvlib>=0.10",
"pytz",
"requests>=2.31.0", # semver
"scikit-learn>=1.1.1", # semver
"scipy>=1.0",
"xlrd>=2",
]
building = [
"cvxpy>=1.3, <2", # semver
"pvlib>=0.10",
"scikit-learn>=1.1.1", # semver
"scipy>=1.0",
]
cogen = [
"openpyxl>=3.1",
"onnxruntime>=1.15, <2", # semver
"openpyxl>=3.1",
"pandas>=1.1, <3", # semver
"xlrd>=2"
"xlrd>=2",
]
building = [
evcharging = [
"acnportal@git+https://github.com/zach401/acnportal@0127a055cf",
"cvxpy>=1.3, <2", # semver
"pvlib>=0.10",
"mosek>=10.0.44, <11", # semver
"pandas>=1.1, <3", # semver
"pytz",
"requests>=2.31.0", # semver
"scikit-learn>=1.1.1", # semver
"scipy>=1.0"
]

[project.urls]
Homepage = "https://chrisyeh96.github.io/sustaingym/"
Repository = "https://github.com/chrisyeh96/sustaingym"
"Bug Tracker" = "https://github.com/chrisyeh96/sustaingym/issues"

[tool.setuptools]
packages = ["sustaingym"]

[tool.setuptools.package-data]
"*" = ["*.csv", "*.csv.gz", "*.epw", "*.htm", "*.json", "*.onnx", "*.pkl", "*.xlsx"]

0 comments on commit d50bdc4

Please sign in to comment.