Skip to content

Comments

Pin numpy to 1.21.2#257

Draft
benjaminrwilson wants to merge 5 commits intomasterfrom
upgrade-numpy
Draft

Pin numpy to 1.21.2#257
benjaminrwilson wants to merge 5 commits intomasterfrom
upgrade-numpy

Conversation

@benjaminrwilson
Copy link
Contributor

@benjaminrwilson benjaminrwilson commented Jul 22, 2021

Draft. Does not need review currently. Will be resolved after disparity_interpolation refactor.

@benjaminrwilson benjaminrwilson marked this pull request as draft July 26, 2021 16:44
self._log_list: Optional[List[str]] = None
self._image_list: Optional[Dict[str, Dict[str, List[str]]]] = None
self._image_list_sync: Optional[Dict[str, Dict[str, List[np.ndarray]]]] = None
self._image_list_sync: Optional[Dict[str, Dict[str, List[NDArray[np.float64]]]]] = None
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hi Anna, the image timestamps should be in nanoseconds (integers), so likely we need to cast to np.int32 or np.int64 somewhere beforehand


@property
def agent_traj(self) -> np.ndarray:
def agent_traj(self) -> Any:
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

maybe we can cast the return arg to np.array() since, np.column_stack() should hopefully only return an array

@benjaminrwilson benjaminrwilson changed the title Pin numpy to 1.2.1 Pin numpy to 1.2.2 Sep 14, 2021
@benjaminrwilson benjaminrwilson changed the title Pin numpy to 1.2.2 Pin numpy to 1.21.2 Sep 14, 2021

assert len(dt_fpaths) == len(gt_fpaths)
data: DefaultDict[str, List[np.ndarray]] = defaultdict(list)
data: DefaultDict[str, List[NDArray[np.float64]]] = defaultdict(list)
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hi @ahuller3, thanks for your work on this.

@benjaminrwilson and I were reviewing this today, and we were thinking it might be easier for readability to do something like

if we create something like argoverse/typing/numpy.py

NDArrayFloat = Union[NDArray[np.float32], NDArray[np.float64]]

What are your thoughts?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants