Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
younik committed Nov 27, 2024
1 parent 5ceb5d9 commit 49f2c19
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 7 deletions.
6 changes: 5 additions & 1 deletion tests/test_cli.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import random

import pytest
from typer.testing import CliRunner

Expand All @@ -9,6 +10,7 @@

runner = CliRunner()


def test_list_app():
result = runner.invoke(app, ["list", "local", "--all"])
assert result.exit_code == 0
Expand Down Expand Up @@ -52,7 +54,9 @@ def test_dataset_download_then_delete(dataset_id: str):
@pytest.mark.parametrize(
"dataset_id",
random.sample(
sorted(list_remote_datasets(compatible_minari_version=True, latest_version=True)),
sorted(
list_remote_datasets(compatible_minari_version=True, latest_version=True)
),
k=8,
),
)
Expand Down
8 changes: 2 additions & 6 deletions tests/test_namespace.py
Original file line number Diff line number Diff line change
Expand Up @@ -121,9 +121,7 @@ def test_download_namespace_dataset():
assert list_local_namespaces() == ["D4RL", "D4RL/kitchen"]
namespace_metadatas = {}
for local_namespace in list_local_namespaces():
namespace_metadatas[local_namespace] = get_namespace_metadata(
local_namespace
)
namespace_metadatas[local_namespace] = get_namespace_metadata(local_namespace)
assert namespace_metadatas[local_namespace] != {}

minari.download_dataset(kitchen_mix)
Expand Down Expand Up @@ -169,7 +167,5 @@ def test_download_namespace_metadata(namespace):
download_namespace_metadata(namespace, overwrite=True)
assert get_namespace_metadata(namespace) != metadata

with pytest.raises(
ValueError, match="doesn't exist in the remote Farama server."
):
with pytest.raises(ValueError, match="doesn't exist in the remote Farama server."):
download_namespace_metadata("non_existent_namespace")

0 comments on commit 49f2c19

Please sign in to comment.