From d389ee270da1eb54b8ee693e9cac1ecb494fcd4f Mon Sep 17 00:00:00 2001
From: Mehran Kholdi <semekh@hamravesh.com>
Date: Mon, 14 Feb 2022 16:48:41 +0330
Subject: [PATCH] Neat: code cleanup

---
 bd2fs.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/bd2fs.py b/bd2fs.py
index 99db366..bc9223a 100644
--- a/bd2fs.py
+++ b/bd2fs.py
@@ -1,7 +1,6 @@
 from pathlib import Path
 
 import grpc
-
 from csi import csi_pb2, csi_pb2_grpc
 from csi.csi_pb2 import (
     NodeStageVolumeRequest,
@@ -30,7 +29,7 @@ def get_fs(request):
 
 
 class Bd2FsIdentityServicer(csi_pb2_grpc.IdentityServicer):
-    def __init__(self, bds):
+    def __init__(self, bds: csi_pb2_grpc.IdentityServicer):
         self.bds = bds
 
     @log_grpc_request
@@ -47,7 +46,7 @@ class Bd2FsIdentityServicer(csi_pb2_grpc.IdentityServicer):
 
 
 class Bd2FsNodeServicer(csi_pb2_grpc.NodeServicer):
-    def __init__(self, bds):
+    def __init__(self, bds: csi_pb2_grpc.NodeServicer):
         self.bds = bds
 
     # @log_grpc_request
@@ -167,7 +166,7 @@ class Bd2FsNodeServicer(csi_pb2_grpc.NodeServicer):
 
 
 class Bd2FsControllerServicer(csi_pb2_grpc.ControllerServicer):
-    def __init__(self, bds):
+    def __init__(self, bds: csi_pb2_grpc.ControllerServicer):
         self.bds = bds
 
     @log_grpc_request