From e10b3688453b1aaf12feb092fb90513595cb28d4 Mon Sep 17 00:00:00 2001 From: Aaron Berdy Date: Mon, 9 Oct 2023 12:04:38 -0700 Subject: [PATCH] fix: address todos --- src/braket/jobs/hybrid_job.py | 7 +++---- test/unit_tests/braket/jobs/test_data_persistence.py | 2 -- 2 files changed, 3 insertions(+), 6 deletions(-) diff --git a/src/braket/jobs/hybrid_job.py b/src/braket/jobs/hybrid_job.py index 2460da6ee..bca0b58cb 100644 --- a/src/braket/jobs/hybrid_job.py +++ b/src/braket/jobs/hybrid_job.py @@ -62,11 +62,11 @@ def hybrid_job( include_modules (Union[Union[str, ModuleType], Iterable[Union[str, ModuleType]]]): Either a single module or module name or a list of module or module names referring to local - modules to be included. Default value `[]` - todo: confirm docs + modules to be included. Any references to members of these modules in the hybrid job + algorithm code will be serialized as part of the algorithm code. Default value `[]` dependencies (Optional[Union[str, Path]]): Path (absolute or relative) to a requirements.txt - file to be used. todo: confirm docs + file to be used for the hybrid job. local (bool): Whether to use local mode for the hybrid job. Default `False` @@ -213,7 +213,6 @@ def __exit__(self, exc_type, exc_val, exc_tb): def _serialize_entry_point(entry_point: Callable, args, kwargs) -> str: """Create an entry point from a function""" - # todo: catch serialization errors def wrapped_entry_point(): """Partial function wrapping entry point with given parameters""" return entry_point(*args, **kwargs) diff --git a/test/unit_tests/braket/jobs/test_data_persistence.py b/test/unit_tests/braket/jobs/test_data_persistence.py index 96a05aecd..6a5e27283 100644 --- a/test/unit_tests/braket/jobs/test_data_persistence.py +++ b/test/unit_tests/braket/jobs/test_data_persistence.py @@ -20,8 +20,6 @@ import numpy as np import pytest -# todo: remove once circular import is fixed -import braket.aws # noqa: F401 from braket.jobs.data_persistence import ( load_job_checkpoint, load_job_result,