fix: Add Latest Change

Signed-off-by: aceppaluni <aceppaluni@gmail.com>

Author: aceppaluni

src/hiero_sdk_python/query/fee_estimate_query.py

@@ -56,6 +56,8 @@ def __init__(self) -> None:
 
     def set_mode(self, mode: FeeEstimateMode) -> FeeEstimateQuery:
         """Set the estimation mode (STATE or INTRINSIC)."""
+        if not isinstance(mode, FeeEstimateMode):
+            raise TypeError("mode must be a FeeEstimateMode")
         self._mode = mode
         return self
 

tests/unit/fee_estimate_query_test.py

@@ -178,6 +178,8 @@ def test_invalid_argument_error(mock_post):
     with pytest.raises(RuntimeError):
         FeeEstimateQuery().set_transaction(tx).execute(mock_client())
 
+    assert mock_post.call_count == 1, "HTTP 400 (INVALID_ARGUMENT) must not be retried"
+
 
 @patch("hiero_sdk_python.query.fee_estimate_query.requests.post")
 def test_retry_on_timeout(mock_post):