Move boto imports

Author: blueraft

src/nomad_ml_workflows/actions/export_remote_entries/activities.py

@@ -9,7 +9,6 @@
 from typing import Any
 from urllib.parse import urlparse
 
-import boto3
 from nomad.actions.manager import action_instance_artifacts_dir
 from nomad.files import StagingUploadFiles
 from nomad.uploads import get_upload_files
@@ -122,6 +121,8 @@ def _upload_dataset_to_s3(
     artifacts_subdirectory: Path,
 ) -> str:
     """Upload exported dataset files to S3-compatible remote storage."""
+    import boto3
+
     client_kwargs = _build_boto3_client_kwargs(storage_settings)
     s3_client = boto3.client('s3', **client_kwargs)
     bucket = storage_settings.bucket
@@ -150,6 +151,8 @@ def _copy_dataset_from_s3_to_upload(
     storage_settings: S3StorageSettings,
 ) -> str:
     """Download an S3 dataset and add it to a NOMAD staging upload."""
+    import boto3
+
     bucket, key = _parse_s3_uri(data.remote_uri)
     client_kwargs = _build_boto3_client_kwargs(storage_settings)
     s3_client = boto3.client('s3', **client_kwargs)

tests/actions/export_remote_entries/test_activities.py

@@ -32,7 +32,7 @@ def test_read_num_entries_exported(mock_artifacts_dir, tmp_path):
 
 
 @pytest.mark.asyncio
-@patch('nomad_ml_workflows.actions.export_remote_entries.activities.boto3.client')
+@patch('boto3.client')
 @patch(
     'nomad_ml_workflows.actions.export_remote_entries.activities.action_instance_artifacts_dir'
 )
@@ -88,7 +88,7 @@ async def test_upload_dataset_to_remote_storage_s3_zipped(
 
 
 @pytest.mark.asyncio
-@patch('nomad_ml_workflows.actions.export_remote_entries.activities.boto3.client')
+@patch('boto3.client')
 @patch(
     'nomad_ml_workflows.actions.export_remote_entries.activities.action_instance_artifacts_dir'
 )
@@ -129,7 +129,7 @@ async def test_upload_dataset_to_remote_storage_s3_unzipped(
 
 
 @pytest.mark.asyncio
-@patch('nomad_ml_workflows.actions.export_remote_entries.activities.boto3.client')
+@patch('boto3.client')
 @patch('nomad_ml_workflows.actions.export_remote_entries.activities.get_upload_files')
 async def test_copy_remote_dataset_to_upload_s3_zipped(
     mock_get_upload_files, mock_boto_client
@@ -164,7 +164,7 @@ async def test_copy_remote_dataset_to_upload_s3_zipped(
 
 
 @pytest.mark.asyncio
-@patch('nomad_ml_workflows.actions.export_remote_entries.activities.boto3.client')
+@patch('boto3.client')
 @patch('nomad_ml_workflows.actions.export_remote_entries.activities.get_upload_files')
 async def test_copy_remote_dataset_to_upload_s3_directory(
     mock_get_upload_files, mock_boto_client