migrated DATASET_SIZE to AMOUNT_OF_CIRCUITS

Author: Dpbm

args/parser.py

@@ -11,7 +11,7 @@
     DEFAULT_MAX_TOTAL_GATES,
     DEFAULT_THREADS,
     DEFAULT_BATCH_SIZE,
-    DEFAULT_DATASET_SIZE,
+    DEFAULT_AMOUNT_OF_CIRCUITS,
     DEFAULT_NEW_DIM,
     DEFAULT_TRAIN_PERCENTAGE,
     DEFAULT_TEST_PERCENTAGE,
@@ -24,6 +24,21 @@
 class Arguments:
     """Parsed args types"""
 
+    __slots__ = [
+        "_epochs", 
+        "_batch_size", 
+        "_train_size", 
+        "_test_size", 
+        "_threads", 
+        "_shots", 
+        "_n_qubits", 
+        "_max_gates", 
+        "_amount_circuits",
+        "_target_folder",
+        "_checkpoint",
+        "_new_image_dim"
+    ]
+
     def __init__(self):
         """set default arguments"""
 
@@ -35,7 +50,7 @@ def __init__(self):
         self._shots = DEFAULT_SHOTS
         self._n_qubits = DEFAULT_NUM_QUBITS
         self._max_gates = DEFAULT_MAX_TOTAL_GATES
-        self._dataset_size = DEFAULT_DATASET_SIZE
+        self._amount_circuits = DEFAULT_AMOUNT_OF_CIRCUITS
         self._target_folder = DEFAULT_TARGET_FOLDER
         self._checkpoint = DEFAULT_CHECKPOINT
         self._new_image_dim = DEFAULT_NEW_DIM
@@ -50,7 +65,7 @@ def parse(self, args: argparse.Namespace):
         self._shots = args.shots
         self._n_qubits = args.n_qubits
         self._max_gates = args.max_gates
-        self._dataset_size = args.dataset_size
+        self._amount_circuits = args.amount_circuits
         self._target_folder = args.target_folder
         self._checkpoint = args.checkpoint
         self._new_image_dim = args.new_image_dim
@@ -136,14 +151,14 @@ def max_gates(self, value: int):
         self._max_gates = value
 
     @property
-    def dataset_size(self) -> int:
-        """Get dataset_size data"""
-        return self._dataset_size  # type: ignore
+    def amount_circuits(self) -> int:
+        """Get amount_circuits data"""
+        return self._amount_circuits  # type: ignore
 
-    @dataset_size.setter
-    def dataset_size(self, value: int):
-        """Set dataset_size data"""
-        self._dataset_size = value
+    @amount_circuits.setter
+    def amount_circuits(self, value: int):
+        """Set amount_circuits data"""
+        self._amount_circuits = value
 
     @property
     def target_folder(self) -> FilePath:
@@ -184,7 +199,7 @@ def __str__(self) -> str:
         string += f"shots: {self._shots}\n"
         string += f"n qubits: {self._n_qubits}\n"
         string += f"max gates: {self._max_gates}\n"
-        string += f"dataset size: {self._dataset_size}\n"
+        string += f"amount circuits: {self._amount_circuits}\n"
         string += f"target_folder: {self._target_folder}\n"
         string += f"checkpoint: {self._checkpoint}\n"
         string += f"new image dim: {self._new_image_dim}\n"
@@ -210,7 +225,7 @@ def parse_args() -> Arguments:
     parser.add_argument("--n-qubits", type=int, default=DEFAULT_NUM_QUBITS)
     parser.add_argument("--max-gates", type=int, default=DEFAULT_MAX_TOTAL_GATES)
 
-    parser.add_argument("--dataset-size", type=int, default=DEFAULT_DATASET_SIZE)
+    parser.add_argument("--amount-circuits", type=int, default=DEFAULT_AMOUNT_OF_CIRCUITS)
     parser.add_argument("--target-folder", type=str, default=DEFAULT_TARGET_FOLDER)
     parser.add_argument("--new-image-dim", type=int, nargs=2, default=DEFAULT_NEW_DIM)
 

dags/dataset.py

@@ -25,7 +25,7 @@
     DEFAULT_NEW_DIM,
     DEFAULT_MAX_TOTAL_GATES,
     DEFAULT_SHOTS,
-    DEFAULT_DATASET_SIZE,
+    DEFAULT_AMOUNT_OF_CIRCUITS,
     DEFAULT_THREADS,
     images_gen_checkpoint_file,
 )
@@ -117,7 +117,7 @@ def update_checkpoint(checkpoint: Checkpoint, stage: Stages):
             DEFAULT_NUM_QUBITS,
             DEFAULT_MAX_TOTAL_GATES,
             DEFAULT_SHOTS,
-            DEFAULT_DATASET_SIZE,
+            DEFAULT_AMOUNT_OF_CIRCUITS,
             DEFAULT_THREADS,
             checkpoint,
         ],

dataset.py

@@ -244,7 +244,7 @@ def generate_images(
     n_qubits: int,
     total_gates: int,
     shots: int,
-    dataset_size: int,
+    amount_circuits: int,
     total_threads: int,
     checkpoint: Checkpoint,
 ):
@@ -270,8 +270,8 @@ def generate_images(
     base_dataset_path = dataset_path(target_folder)
 
     index = checkpoint.index
-    with tqdm(total=dataset_size, initial=index) as progress:
-        while index < dataset_size:
+    with tqdm(total=amount_circuits, initial=index) as progress:
+        while index < amount_circuits:
             args = []
 
             for i in range(total_threads):
@@ -477,7 +477,7 @@ def main(args: Arguments):
             args.n_qubits,
             args.max_gates,
             args.shots,
-            args.dataset_size,
+            args.amount_circuits,
             args.threads,
             checkpoint,
         )

tests/test_args.py

@@ -4,7 +4,7 @@
 from utils.constants import (
     DEFAULT_BATCH_SIZE,
     DEFAULT_CHECKPOINT,
-    DEFAULT_DATASET_SIZE,
+    DEFAULT_AMOUNT_OF_CIRCUITS,
     DEFAULT_EPOCHS,
     DEFAULT_MAX_TOTAL_GATES,
     DEFAULT_NEW_DIM,
@@ -39,7 +39,7 @@ def test_default_values(self):
 
         assert args.batch_size == DEFAULT_BATCH_SIZE
         assert args.checkpoint == DEFAULT_CHECKPOINT
-        assert args.dataset_size == DEFAULT_DATASET_SIZE
+        assert args.amount_circuits == DEFAULT_AMOUNT_OF_CIRCUITS
         assert args.epochs == DEFAULT_EPOCHS
         assert args.max_gates == DEFAULT_MAX_TOTAL_GATES
         assert args.n_qubits == DEFAULT_NUM_QUBITS
@@ -71,7 +71,7 @@ def test_arbitrary_arguments(self):
             "6",
             "--max-gates",
             "103",
-            "--dataset-size",
+            "--amount-circuits",
             "3233",
             "--target-folder",
             "another",
@@ -91,6 +91,6 @@ def test_arbitrary_arguments(self):
         assert args.shots == int(sys.argv[14])
         assert args.n_qubits == int(sys.argv[16])
         assert args.max_gates == int(sys.argv[18])
-        assert args.dataset_size == int(sys.argv[20])
+        assert args.amount_circuits == int(sys.argv[20])
         assert args.target_folder == sys.argv[22]
         assert args.new_image_dim == [int(sys.argv[24]), int(sys.argv[25])]

utils/constants.py

@@ -14,7 +14,7 @@
 
 DEFAULT_BATCH_SIZE = 10
 
-DEFAULT_DATASET_SIZE = 20000 # this one doesn't reflect exactly the size of the dataset, once the dataset might get either bigger, due to the different combinations of mesurements, or smaller due to duplicated circuits
+DEFAULT_AMOUNT_OF_CIRCUITS = 20000 # this one doesn't reflect exactly the size of the dataset, once the dataset might get either bigger, due to the different combinations of mesurements, or smaller due to duplicated circuits
 
 DEFAULT_TARGET_FOLDER = "."