code: def test_model_wo_softmax_functional():
      inputs = klayers.Input(shape=(10,))
      outputs = klayers.Dense(10, activation="softmax")(inputs)
      model = kmodels.Model(inputs=inputs, outputs=outputs)
      model = innvestigate.model_wo_softmax(model)
      assert model.layers[-1].activation.__name__ == "linear"
  
file_path: innvestigate/tests/backend/test_model_wo_softmax.py
imports: from __future__ import annotations
  import pytest
  import tensorflow.keras.layers
  import tensorflow.keras.models
  import innvestigate
label_desc: Use image node_id 2 for calling a global function with example usage: test_model_wo_softmax_functional() without return types