From d087931131e055dc42296026e70b313dad889e22 Mon Sep 17 00:00:00 2001 From: samkoebrich Date: Thu, 13 Feb 2025 12:06:58 -0700 Subject: [PATCH 1/2] set max_workers explictily --- watttime/api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/watttime/api.py b/watttime/api.py index b70da778..cc8207e2 100644 --- a/watttime/api.py +++ b/watttime/api.py @@ -258,7 +258,7 @@ def _fetch_data( responses = [] if self.multithreaded: - with ThreadPoolExecutor(max_workers=os.cpu_count() * 5) as executor: + with ThreadPoolExecutor(max_workers=10) as executor: futures = { executor.submit( self._make_rate_limited_request, url, params From 95e8df5ac7a50e599bffbf7bc49e4bc3a2a69894 Mon Sep 17 00:00:00 2001 From: samkoebrich Date: Thu, 13 Feb 2025 16:39:19 -0700 Subject: [PATCH 2/2] don't use session in test (yet) --- tests/test_sdk.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_sdk.py b/tests/test_sdk.py index f8bd74ca..220883f3 100644 --- a/tests/test_sdk.py +++ b/tests/test_sdk.py @@ -135,7 +135,7 @@ def test_parse_dates_with_datetime(self): self.assertIsInstance(parsed_end, datetime) self.assertEqual(parsed_end.tzinfo, UTC) - @mock.patch("watttime.requests.Session.post", side_effect=mocked_register) + @mock.patch("requests.post", side_effect=mocked_register) def test_mock_register(self, mock_post): resp = self.base.register(email=os.getenv("WATTTIME_EMAIL")) self.assertEqual(len(mock_post.call_args_list), 1)