Skip to content

Commit

Permalink
Merge pull request #276 from ihmeuw/collijk/refactor/address-randomne…
Browse files Browse the repository at this point in the history
…ss-pr-comments

Address PR comments
  • Loading branch information
collijk committed Feb 22, 2023
2 parents 9433869 + 63fdba2 commit 5c315b5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/vivarium/framework/randomness/index_map.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ class IndexMap:

def __init__(self, use_crn: bool = True, size: int = 1_000_000):
self._use_crn = use_crn
self._map = pd.Series(dtype=float)
self._map = pd.Series(dtype=int)
self._size = size

def update(self, new_keys: pd.Index) -> None:
Expand Down Expand Up @@ -71,7 +71,7 @@ def _hash(self, keys: pd.Index, salt: int = 0) -> pd.Series:
-------
pandas.Series
A pandas series indexed by the given keys and whose values take on
integers in the range [0, self.stride]. Duplicates may appear and
integers in the range [0, len(self)]. Duplicates may appear and
should be dealt with by the calling code.
"""
Expand Down

0 comments on commit 5c315b5

Please sign in to comment.