small fix transferlearn set output activation

aertslab · Sep 23, 2024 · 3e0d53a · 3e0d53a
1 parent ac84f35
commit 3e0d53a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/crested/tl/_crested.py b/src/crested/tl/_crested.py
@@ -447,7 +447,7 @@ def transferlearn(
         old_activation_layer = base_model.layers[-1]
         if set_output_activation is None:
             old_activation = old_activation_layer.activation
-        else set_output_activation:
+        elif set_output_activation:
             old_activation = keras.activations.get(set_output_activation)
 
         x = base_model.layers[-3].output