Skip to content

Commit

Permalink
Update clf_analysis_tool.py
Browse files Browse the repository at this point in the history
Add type hint comment to partial_model
  • Loading branch information
dbkchu committed Jul 19, 2024
1 parent ffa4caf commit 3e01df3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion molSimplify/python_nn/clf_analysis_tool.py
Original file line number Diff line number Diff line change
Expand Up @@ -105,7 +105,7 @@ def get_layer_outputs(model, layer_index, input,
[model.layers[layer_index].output])
nn_outputs = get_outputs([input, training_flag])[0]
else:
partial_model = Model(model.inputs, model.layers[layer_index].output)
partial_model = Model(model.inputs, model.layers[layer_index].output) # type: Callable
nn_outputs = partial_model([input], training=training_flag).numpy() # runs the model in training mode
return nn_outputs

Expand Down

0 comments on commit 3e01df3

Please sign in to comment.