Skip to content

Commit

Permalink
black
Browse files Browse the repository at this point in the history
  • Loading branch information
ataymano committed Nov 15, 2023
1 parent c333c0a commit 56d69f0
Showing 1 changed file with 12 additions and 3 deletions.
15 changes: 12 additions & 3 deletions tests/unit_tests/test_rl_loop_base_embedder.py
Original file line number Diff line number Diff line change
Expand Up @@ -362,9 +362,18 @@ def test_action_w_namespace_w_emb_w_more_than_one_item_in_first_dict() -> None:
assert featurized[i].dense == expected_dense[i]

expected_sparse = [
{"test_namespace": {"default_ft": str1}, "test_namespace2": {"default_ft": str1}},
{"test_namespace": {"default_ft": str2}, "test_namespace2": {"default_ft": str2}},
{"test_namespace": {"default_ft": str3}, "test_namespace2": {"default_ft": str3}},
{
"test_namespace": {"default_ft": str1},
"test_namespace2": {"default_ft": str1},
},
{
"test_namespace": {"default_ft": str2},
"test_namespace2": {"default_ft": str2},
},
{
"test_namespace": {"default_ft": str3},
"test_namespace2": {"default_ft": str3},
},
]
featurized = base.embed(
[
Expand Down

0 comments on commit 56d69f0

Please sign in to comment.