Skip to content

Commit 140dbb0

Browse files
committed
chore: address PR feedback
Signed-off-by: dosi <[email protected]>
1 parent fd318a4 commit 140dbb0

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

src/hiero_sdk_python/system/freeze_transaction.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
from hiero_sdk_python.hapi.services.schedulable_transaction_body_pb2 import (
1212
SchedulableTransactionBody,
1313
)
14+
from hiero_sdk_python.hapi.services.transaction_pb2 import TransactionBody
1415
from hiero_sdk_python.system.freeze_type import FreezeType
1516
from hiero_sdk_python.timestamp import Timestamp
1617
from hiero_sdk_python.transaction.transaction import Transaction
@@ -119,7 +120,7 @@ def _build_proto_body(self) -> FreezeTransactionBody:
119120
freeze_type=self.freeze_type._to_proto() if self.freeze_type else None,
120121
)
121122

122-
def build_transaction_body(self):
123+
def build_transaction_body(self) -> TransactionBody:
123124
"""
124125
Builds the transaction body for this freeze transaction.
125126

src/hiero_sdk_python/system/freeze_type.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ class FreezeType(Enum):
2828
TELEMETRY_UPGRADE = 5
2929

3030
@staticmethod
31-
def _from_proto(proto_obj: proto_FreezeType):
31+
def _from_proto(proto_obj: proto_FreezeType) -> "FreezeType":
3232
"""
3333
Converts a protobuf FreezeType to a FreezeType enum.
3434
@@ -72,8 +72,9 @@ def _to_proto(self) -> proto_FreezeType:
7272
return proto_FreezeType.TELEMETRY_UPGRADE
7373
return proto_FreezeType.UNKNOWN_FREEZE_TYPE
7474

75-
def __eq__(self, other):
75+
def __eq__(self, other: object) -> bool:
7676
if isinstance(other, FreezeType):
7777
return self.value == other.value
7878
elif isinstance(other, int):
7979
return self.value == other
80+
return False

0 commit comments

Comments
 (0)