[minor] minor fix

sequence-toolbox · Sep 13, 2024 · 63da9e0 · 63da9e0
1 parent 31c26d0
commit 63da9e0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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')