Skip to content

Commit

Permalink
Added an option to provide seed for random number generator
Browse files Browse the repository at this point in the history
  • Loading branch information
arindamsaha1507 committed Feb 26, 2024
1 parent 0afab67 commit 2836019
Showing 1 changed file with 17 additions and 0 deletions.
17 changes: 17 additions & 0 deletions run.py
Expand Up @@ -2,10 +2,13 @@

import argparse
import csv
import random
import sys
from datetime import datetime, timedelta
from os import makedirs, path

import numpy as np

from facs.base import facs
from facs.base.measures import Measures
from facs.readers import (
Expand Down Expand Up @@ -119,6 +122,15 @@ def parse_arguments() -> dict:
default="20",
help="Workspace per person in m2.",
)

parser.add_argument(
"--seed",
action="store",
type=int,
default=None,
help="Seed for random number generator.",
)

return parser.parse_args()


Expand Down Expand Up @@ -166,6 +178,11 @@ def main():

print(args)

if args.seed is not None:
np.random.seed(args.seed)
random.seed(args.seed)
print(f"Random seed set to {args.seed}")

house_ratio = get_house_ratio(args.quicktest)
location = args.location
measures_yml = get_measures_file(args.measures_yml)
Expand Down

0 comments on commit 2836019

Please sign in to comment.