Merge pull request #9449 from freqtrade/dependabot/pip/develop/torch-2.1.1

Bump torch from 2.0.1 to 2.1.1
This commit is contained in:
Robert Caulk 2023-12-10 14:11:01 +01:00 committed by GitHub
commit c0119d7baa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 16 additions and 1 deletions

View File

@ -2,7 +2,7 @@
-r requirements-freqai.txt
# Required for freqai-rl
torch==2.0.1
torch==2.1.1
#until these branches will be released we can use this
gymnasium==0.29.1
stable_baselines3==2.2.1

View File

@ -20,6 +20,21 @@ def is_mac() -> bool:
return "Darwin" in machine
@pytest.fixture(autouse=True)
def patch_torch_initlogs(mocker) -> None:
if is_mac():
# Mock torch import completely
import sys
import types
module_name = 'torch'
mocked_module = types.ModuleType(module_name)
sys.modules[module_name] = mocked_module
else:
mocker.patch("torch._logging._init_logs")
@pytest.fixture(scope="function")
def freqai_conf(default_conf, tmp_path):
freqaiconf = deepcopy(default_conf)