Skip to content

Commit 54c834f

Browse files
committed
again hide functional modules
#21
1 parent 80009e5 commit 54c834f

File tree

2 files changed

+143
-143
lines changed

2 files changed

+143
-143
lines changed

nn/_generate_layers.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -826,7 +826,7 @@ def get_module_class_name_for_layer_class(sig: LayerSignature) -> str:
826826
name = name[:-len("Layer")]
827827
if name.startswith("_"):
828828
return name
829-
if layer_class.layer_class in LayersHidden:
829+
if layer_class.layer_class in LayersHidden or sig.is_functional():
830830
return "_" + name # we make a public function for it, but the module is hidden
831831
return name
832832

0 commit comments

Comments
 (0)