fixed use of pipe operator.

RasaHQ · Oct 4, 2023 · a088531 · a088531
1 parent 6fb3dd8
commit a088531
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/dialogue_understanding/generator/test_llm_command_generator.py b/tests/dialogue_understanding/generator/test_llm_command_generator.py
@@ -1,4 +1,4 @@
-from typing import Optional
+from typing import Optional, Any
 from unittest.mock import Mock, patch
 
 import pytest
@@ -233,8 +233,8 @@ def test_coerce_slot_value(
         self,
         slot_name: str,
         slot: Slot,
-        slot_value: Optional[str | int | float | bool],
-        expected_output: Optional[str | int | float | bool],
+        slot_value: Any,
+        expected_output: Any,
     ):
         """Test that coerce_slot_value coerces the slot value correctly."""
         # Given