Skip to content

Commit

Permalink
Style
Browse files Browse the repository at this point in the history
  • Loading branch information
freddyaboulton committed Jul 26, 2023
1 parent 6ddca1b commit 440cda8
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tests/test_hf_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -2524,7 +2524,7 @@ def setUp(self) -> None:

get_session_mock = Mock()
self.post_mock = get_session_mock().post
self.post_mock.return_value.json.return_value = {
self.post_mock.return_value.json.return_value = {
"url": f"{self.api.endpoint}/spaces/user/repo_id",
"stage": "RUNNING",
"sdk": "gradio",
Expand All @@ -2537,7 +2537,7 @@ def setUp(self) -> None:
"gcTimeout": None,
}
self.delete_mock = get_session_mock().delete
self.delete_mock.return_value.json.return_value = {
self.delete_mock.return_value.json.return_value = {
"url": f"{self.api.endpoint}/spaces/user/repo_id",
"stage": "RUNNING",
"sdk": "gradio",
Expand Down Expand Up @@ -2603,7 +2603,7 @@ def test_set_space_sleep_time_cpu_basic(self) -> None:
self.post_mock.return_value.json.return_value["hardware"]["requested"] = "cpu-basic"
with self.assertWarns(UserWarning):
self.api.set_space_sleep_time(self.repo_id, sleep_time=123)

def test_request_space_storage(self) -> None:
runtime = self.api.request_space_storage(self.repo_id, SpaceStorage.LARGE)
self.post_mock.assert_called_once_with(
Expand All @@ -2612,7 +2612,7 @@ def test_request_space_storage(self) -> None:
json={"tier": "large"},
)
assert runtime.storage == SpaceStorage.LARGE

def test_delete_space_storage(self) -> None:
runtime = self.api.delete_space_storage(self.repo_id)
self.delete_mock.assert_called_once_with(
Expand Down

0 comments on commit 440cda8

Please sign in to comment.