Skip to content

Commit e6d825e

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent d392e21 commit e6d825e

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

dynamic_programming/viterbi.py

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,12 @@
33

44

55
def viterbi(
6-
observations_space: List[str],
7-
states_space: List[str],
8-
initial_probabilities: Dict[str, float],
9-
transition_probabilities: Dict[str, Dict[str, float]],
10-
emission_probabilities: Dict[str, Dict[str, float]],
11-
) -> List[str]:
6+
observations_space: list[str],
7+
states_space: list[str],
8+
initial_probabilities: dict[str, float],
9+
transition_probabilities: dict[str, dict[str, float]],
10+
emission_probabilities: dict[str, dict[str, float]],
11+
) -> list[str]:
1212
"""
1313
Viterbi Algorithm, to find the most likely path of
1414
states from the start and the expected output.
@@ -246,11 +246,11 @@ def _validate_dict(_object: Any, var_name: str, value_type: type, nested: bool =
246246

247247

248248
def _initialise_probabilities_and_pointers(
249-
observations_space: List[str],
250-
states_space: List[str],
251-
initial_probabilities: Dict[str, float],
252-
emission_probabilities: Dict[str, Dict[str, float]],
253-
) -> Tuple[dict, dict]:
249+
observations_space: list[str],
250+
states_space: list[str],
251+
initial_probabilities: dict[str, float],
252+
emission_probabilities: dict[str, dict[str, float]],
253+
) -> tuple[dict, dict]:
254254
probabilities = {}
255255
pointers = {}
256256
for state in states_space:
@@ -263,8 +263,8 @@ def _initialise_probabilities_and_pointers(
263263

264264

265265
def _process_forward(
266-
observations_space: List[str],
267-
states_space: List[str],
266+
observations_space: list[str],
267+
states_space: list[str],
268268
_prior_state: Callable,
269269
probabilities: dict,
270270
pointers: dict,
@@ -294,10 +294,10 @@ def _best_final_state(k_state: str) -> float:
294294

295295

296296
def _extract_best_path(
297-
observations_space: List[str],
297+
observations_space: list[str],
298298
last_observation: str,
299299
pointers: dict,
300-
) -> List[str]:
300+
) -> list[str]:
301301
previous = last_observation
302302
result = []
303303
for o in range(len(observations_space) - 1, -1, -1):
@@ -307,7 +307,7 @@ def _extract_best_path(
307307
return result
308308

309309

310-
def _arg_max(prior_state: Callable, states_space: List[str]) -> str:
310+
def _arg_max(prior_state: Callable, states_space: list[str]) -> str:
311311
arg_max = ""
312312
max_probability = -1
313313
for k_state in states_space:

0 commit comments

Comments
 (0)