Skip to content

Commit 2fb3f92

Browse files
committed
Add type hinting on iterable
1 parent 11a40c3 commit 2fb3f92

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

avapi/_dataset.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import json
22
import os
33
import random
4-
from typing import Tuple, Union
4+
from typing import Iterable, Tuple, Union
55

66
import numpy as np
77
from avstack import calibration, sensors
@@ -44,7 +44,7 @@ def get_reference_from_line(line):
4444

4545

4646
class BaseSceneManager:
47-
def __iter__(self):
47+
def __iter__(self) -> Iterable["BaseSceneDataset"]:
4848
for scene in self.scenes:
4949
yield self.get_scene_dataset_by_name(scene)
5050

0 commit comments

Comments
 (0)