fixes linting

WildMeOrg · Jun 13, 2024 · d920cce · d920cce
1 parent 5b2322a
commit d920cce
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wbia/algo/detect/efficientnet.py b/wbia/algo/detect/efficientnet.py
@@ -169,7 +169,7 @@ def __init__(self, n_class, model_arch='tf_efficientnet_b4_ns', pretrained=False
         if n_class is not None:
             n_features = self.model.classifier.in_features
             self.model.classifier = nn.Linear(n_features, n_class)
-            
+
         else:
             self.model.classifier = nn.Identity(n_features, n_class)
         '''
@@ -179,7 +179,7 @@ def __init__(self, n_class, model_arch='tf_efficientnet_b4_ns', pretrained=False
             nn.Linear(n_features, n_class, bias=True)
         )
         '''
-        
+
     def process_row(self, row_labels, preds, sort_weights, labels):
         multi_labels = labels[row_labels.astype(bool)]
         preds = preds[row_labels.astype(bool)]