Skip to content

Commit 03880ed

Browse files
committed
Missing code annotations
1 parent 93599dd commit 03880ed

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

examples/formbot/actions.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,9 @@ def slot_mappings(self) -> Dict[Text, Union[Dict, List[Dict]]]:
2929
return {
3030
"cuisine": self.from_entity(entity="cuisine", not_intent="chitchat"),
3131
"num_people": [
32-
self.from_entity(entity="number", intent=["inform", "request_restaurant"]),
32+
self.from_entity(
33+
entity="number", intent=["inform", "request_restaurant"]
34+
),
3335
],
3436
"outdoor_seating": [
3537
self.from_entity(entity="seating"),

rasa/nlu/training_data/training_data.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ def merge(self, *others: "TrainingData") -> "TrainingData":
7575
nlg_stories,
7676
)
7777

78-
def filter_training_examples(self, condition: Callable[[Message], bool]):
78+
def filter_training_examples(self, condition: Callable[[Message], bool]) -> "TrainingData":
7979
"""Filter training examples.
8080
8181
Args:
@@ -94,15 +94,15 @@ def filter_training_examples(self, condition: Callable[[Message], bool]):
9494
self.lookup_tables,
9595
)
9696

97-
def filter_by_intent(self, intent: Text):
97+
def filter_by_intent(self, intent: Text) -> "TrainingData":
9898
"""Filter training examples."""
9999
raise_warning(
100100
"The `filter_by_intent` function is deprecated. "
101101
"Please use `filter_training_examples` instead.",
102102
DeprecationWarning,
103103
stacklevel=2,
104104
)
105-
self.filter_training_examples(
105+
return self.filter_training_examples(
106106
lambda ex: intent == ex.get("intent")
107107
)
108108

0 commit comments

Comments
 (0)