diff --git a/sequence/topology/node.py b/sequence/topology/node.py index e28c2762..afc7de0d 100644 --- a/sequence/topology/node.py +++ b/sequence/topology/node.py @@ -74,7 +74,7 @@ def init(self) -> None: def set_seed(self, seed: int) -> None: self.generator = np.random.default_rng(seed) - def get_generator(self): + def get_generator(self) -> np.random.Generator: return self.generator def add_component(self, component: Entity) -> None: diff --git a/sequence/utils/config_generator.py b/sequence/utils/config_generator.py index 2a835db3..4c867c59 100644 --- a/sequence/utils/config_generator.py +++ b/sequence/utils/config_generator.py @@ -19,7 +19,7 @@ def add_default_args(parser): """ parser.add_argument('memo_size', type=int, help='number of memories per node') - parser.add_argument('qc_length', type=float, help='distance between nodes (in m)') + parser.add_argument('qc_length', type=float, help='distance between nodes (in km)') parser.add_argument('qc_atten', type=float, help='quantum channel attenuation (in dB/m)') parser.add_argument('cc_delay', type=float, help='classical channel delay (in ms)') parser.add_argument('-d', '--directory', type=str, default='tmp', help='name of output directory')