From 545f18053f0e2ced7654d2126082533e1e444f29 Mon Sep 17 00:00:00 2001
From: marquessv <marquessavaldez@gmail.com>
Date: Thu, 10 Aug 2023 15:43:48 -0700
Subject: [PATCH] a few more cleanups

---
 pyquil/api/_qpu.py | 5 +----
 pyquil/api/_qvm.py | 7 +++----
 2 files changed, 4 insertions(+), 8 deletions(-)

diff --git a/pyquil/api/_qpu.py b/pyquil/api/_qpu.py
index 77f655a74..288ac747e 100644
--- a/pyquil/api/_qpu.py
+++ b/pyquil/api/_qpu.py
@@ -112,7 +112,6 @@ class QPUExecuteResponse:
     job_id: str
     _executable: EncryptedProgram
     execution_options: Optional[ExecutionOptions]
-    _memory_map: MemoryMap
 
 
 class QPU(QAM[QPUExecuteResponse]):
@@ -198,9 +197,7 @@ def execute(
             execution_options=execution_options or self.execution_options,
         )
 
-        return QPUExecuteResponse(
-            _executable=executable, job_id=job_id, execution_options=execution_options, _memory_map=memory_map
-        )
+        return QPUExecuteResponse(_executable=executable, job_id=job_id, execution_options=execution_options)
 
     def get_result(self, execute_response: QPUExecuteResponse) -> QAMExecutionResult:
         """
diff --git a/pyquil/api/_qvm.py b/pyquil/api/_qvm.py
index d837b111d..16491469a 100644
--- a/pyquil/api/_qvm.py
+++ b/pyquil/api/_qvm.py
@@ -14,8 +14,9 @@
 #    limitations under the License.
 ##############################################################################
 from dataclasses import dataclass
-from typing import Any, Optional, Sequence, Tuple
+from typing import Any, Optional, Sequence, Tuple, Dict
 
+import numpy as np
 from qcs_sdk import QCSClient, qvm, ResultData, ExecutionData
 from qcs_sdk.qvm import QVMOptions, QVMResultData
 
@@ -52,7 +53,7 @@ class QVMExecuteResponse:
     data: QVMResultData
 
     @property
-    def memory(self):
+    def memory(self) -> Dict[str, np.ndarray]:
         register_map = self.data.to_register_map()
         return {key: matrix.to_ndarray() for key, matrix in register_map.items()}
 
@@ -158,8 +159,6 @@ def execute(
     def get_result(self, execute_response: QVMExecuteResponse) -> QAMExecutionResult:
         """
         Return the results of execution on the QVM.
-
-        Because QVM execution is synchronous, this is a no-op which returns its input.
         """
 
         result_data = ResultData(execute_response.data)