Skip to content

Commit

Permalink
fix test
Browse files Browse the repository at this point in the history
  • Loading branch information
eelcovdw committed Jun 18, 2024
1 parent 9395e0c commit db1e59b
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -156,7 +156,7 @@ def compute() -> int:

_ = client_low_ds.code.request_code_execution(compute)

diff = compare_clients(low_client, high_client)
diff = compare_clients(low_client, high_client, hide_usercode=False)

assert len(diff.batches) == 2 # Request + UserCode
assert len(diff.ignored_batches) == 0
Expand All @@ -165,7 +165,7 @@ def compute() -> int:
res = diff[0].ignore()
assert isinstance(res, SyftSuccess)

diff = compare_clients(low_client, high_client)
diff = compare_clients(low_client, high_client, hide_usercode=False)
assert len(diff.batches) == 0
assert len(diff.ignored_batches) == 2
assert len(diff.all_batches) == 2
Expand All @@ -174,7 +174,7 @@ def compute() -> int:
res = diff.ignored_batches[0].unignore()
assert isinstance(res, SyftSuccess)

diff = compare_clients(low_client, high_client)
diff = compare_clients(low_client, high_client, hide_usercode=False)
assert len(diff.batches) == 1
assert len(diff.ignored_batches) == 1
assert len(diff.all_batches) == 2
Expand Down

0 comments on commit db1e59b

Please sign in to comment.