Skip to content

Commit

Permalink
fix numpy import error
Browse files Browse the repository at this point in the history
  • Loading branch information
ancalita committed Oct 9, 2023
1 parent 32b0587 commit 850b30e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions rasa/plugin.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,14 +6,14 @@
import pluggy

from rasa.cli import SubParsersAction
from rasa.engine.storage.storage import ModelMetadata
from rasa.shared.core.trackers import DialogueStateTracker
from rasa.shared.nlu.training_data.message import Message

if TYPE_CHECKING:
from rasa.core.brokers.broker import EventBroker
from rasa.core.tracker_store import TrackerStore
from rasa.engine.graph import SchemaNode
from rasa.engine.storage.storage import ModelMetadata
from rasa.shared.core.domain import Domain
from rasa.utils.endpoints import EndpointConfig

Expand Down Expand Up @@ -95,7 +95,7 @@ def init_telemetry(endpoints_file: Optional[Text]) -> None:

@hookspec # type: ignore[misc]
def mock_tracker_for_evaluation(
example: Message, model_metadata: Optional[ModelMetadata]
example: Message, model_metadata: Optional["ModelMetadata"]
) -> Optional[DialogueStateTracker]:
"""Generate a mocked tracker for NLU evaluation."""

Expand Down

0 comments on commit 850b30e

Please sign in to comment.