From 0d307facb16236174c302dc7714f7f20f434a4a6 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Thu, 31 Dec 2020 10:57:35 +0800
Subject: Make super class constructor revoke consistent

---
 models/layers.py | 2 +-
 utils/dataset.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/models/layers.py b/models/layers.py
index c69ae07..a9f04b3 100644
--- a/models/layers.py
+++ b/models/layers.py
@@ -154,7 +154,7 @@ class BasicConv1d(nn.Module):
             kernel_size: Union[int, tuple[int]],
             **kwargs
     ):
-        super(BasicConv1d, self).__init__()
+        super().__init__()
         self.conv = nn.Conv1d(in_channels, out_channels, kernel_size,
                               bias=False, **kwargs)
 
diff --git a/utils/dataset.py b/utils/dataset.py
index bb4d762..ded9fd5 100644
--- a/utils/dataset.py
+++ b/utils/dataset.py
@@ -54,7 +54,7 @@ class CASIAB(data.Dataset):
             num_input_channels = 3, frame_height = 64, frame_width = 32)
             need about 22 GB of RAM.
         """
-        super(CASIAB, self).__init__()
+        super().__init__()
         self._root_dir = root_dir
         self._is_train = is_train
         self._num_sampled_frames = num_sampled_frames
-- 
cgit v1.2.3