code: def test_model_wo_softmax_sequential():
      model = kmodels.Sequential(
          [klayers.Dense(10, input_shape=(10,), activation="softmax")]
      )
      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 1 for calling a global function with example usage: test_model_wo_softmax_sequential() without return types