Prevent double 'Proof' from method and type

Author: Tkd-Alex

src/sentinel_sdk/modules/session.py

@@ -5,7 +5,7 @@
 import sentinel_protobuf.sentinel.session.v2.querier_pb2_grpc as sentinel_session_v2_querier_pb2_grpc
 import sentinel_protobuf.sentinel.session.v2.msg_pb2 as msg_pb2
 
-from sentinel_protobuf.sentinel.session.v2.proof_pb2 import Proof
+from sentinel_protobuf.sentinel.session.v2 import proof_pb2
 from sentinel_protobuf.sentinel.types.v1.bandwidth_pb2 import Bandwidth
 from google.protobuf.duration_pb2 import Duration
 
@@ -100,16 +100,16 @@ def EndSession(self, session_id: int, rating: int, tx_params: TxParams = TxParam
         )
         return self.transaction([msg], tx_params)
 
-    def UpdateDetails(self, proof: Proof, signature: bytes, tx_params: TxParams = TxParams()):
+    def UpdateDetails(self, proof: proof_pb2.Proof, signature: bytes, tx_params: TxParams = TxParams()):
         msg = msg_pb2.MsgUpdateDetailsRequest(
             frm = self._account.address,
             proof = proof,
             signature = signature,
         )
         return self.transaction([msg], tx_params)
 
-    def Proof(self, session_id: int, bandwidth: Bandwidth, duration: Duration) -> Proof:
-        return Proof(
+    def Proof(self, session_id: int, bandwidth: Bandwidth, duration: Duration) -> proof_pb2.Proof:
+        return proof_pb2.Proof(
             id=session_id,
             bandwidth=bandwidth,
             duration=duration