fix: Fix Changes

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

Author: aceppaluni

src/hiero_sdk_python/query/fee_estimate_query.py

@@ -67,6 +67,10 @@ def get_mode(self) -> FeeEstimateMode | None:
 
     def set_transaction(self, transaction: Transaction) -> FeeEstimateQuery:
         """Attach the transaction to estimate."""
+        from hiero_sdk_python.transaction.transaction import Transaction
+
+        if not isinstance(transaction, Transaction):
+            raise TypeError(f"transaction must be an instance of Transaction, got {type(transaction).__name__}")
         self._transaction = transaction
         return self
 

tests/integration/fee_estimate_query_e2e_test.py

@@ -33,7 +33,6 @@ def test_can_execute_fee_estimation_query2(env):
     query = FeeEstimateQuery().set_mode(FeeEstimateMode.STATE).set_transaction(tx)
     result = query.execute(env.client)
 
-    print(type(result.total))
     assert result is not None