tommy/libpostal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[fix] floor samples

Browse Source
This commit is contained in:
Al
2016-05-24 11:16:57 -04:00
parent e53e61358d
commit ad4b197ead
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/geodata/addresses/floors.py
View File

@@ -26,14 +26,14 @@ class Floor(NumberedComponent):
@classmethod @classmethod
def sample_floors(cls, num_floors, num_basements=0): def sample_floors(cls, num_floors, num_basements=0):
num_floors = int(num_floors) num_floors = int(num_floors)
return random.randint(-num_basements, num_floors - 1) return random.randint(-num_basements, (num_floors - 1) if num_floors > 0 else 0)
@classmethod @classmethod
def sample_positive_floors(cls, num_floors, zeroth_floor_prob=0.001): def sample_positive_floors(cls, num_floors, zeroth_floor_prob=0.001):
num_floors = int(num_floors) num_floors = int(num_floors)
if random.random() < zeroth_floor_prob: if random.random() < zeroth_floor_prob:
return 0 return 0
return random.randint(1, num_floors - 1) return random.randint(1, (num_floors - 1) if num_floors > 1 else 1)
@classmethod @classmethod
def random(cls, language, country=None, num_floors=None, num_basements=None): def random(cls, language, country=None, num_floors=None, num_basements=None):