typo

secondmind-labs · Jun 19, 2024 · 1be1c67 · 1be1c67
1 parent 3d9143a
commit 1be1c67
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/gpflux/layers/test_trackable_layer.py b/tests/gpflux/layers/test_trackable_layer.py
@@ -34,7 +34,7 @@ def __init__(self, attributes):
         super().__init__()
 
 
-class UntrackableCompositeLayer(tf_keras.Layer):
+class UntrackableCompositeLayer(tf_keras.layers.Layer):
     def __init__(self, attributes):
         for i, a in enumerate(attributes):
             setattr(self, f"var_{i}", a)