Skip to content

Commit

Permalink
Resolve dependency problems (#511)
Browse files Browse the repository at this point in the history
* Add a temporary upper bounds on scipy and orbax-checkpoint versions

* Resolve ray import issue
  • Loading branch information
bwohlberg authored Apr 9, 2024
1 parent d9026d3 commit 43b172c
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
typing_extensions
numpy>=1.20.0
scipy>=1.6.0
scipy>=1.6.0,<1.13
imageio>=2.17
tifffile
matplotlib
jaxlib>=0.4.3,<=0.4.23
jax>=0.4.3,<=0.4.23
orbax-checkpoint
orbax-checkpoint<=0.5.7
flax>=0.6.1,<=0.7.5
pyabel>=0.9.0
4 changes: 2 additions & 2 deletions scico/ray/tune.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
# -*- coding: utf-8 -*-
# Copyright (C) 2021-2023 by SCICO Developers
# Copyright (C) 2021-2024 by SCICO Developers
# All rights reserved. BSD 3-clause License.
# This file is part of the SCICO package. Details of the copyright and
# user license can be found in the 'LICENSE' file distributed with the
Expand All @@ -23,7 +23,7 @@
except ImportError:
raise ImportError("Could not import ray.tune; please install it.")
import ray.air
from ray.tune import Trainable, loguniform, report, uniform, with_parameters # noqa
from ray.tune import Trainable, loguniform, uniform, with_parameters # noqa
from ray.tune.experiment.trial import Trial
from ray.tune.progress_reporter import TuneReporterBase, _get_trials_by_state
from ray.tune.result_grid import ResultGrid
Expand Down

0 comments on commit 43b172c

Please sign in to comment.