From e11e67ff590976b451a0d3a9705978d9221ddc57 Mon Sep 17 00:00:00 2001 From: Stephen Polcyn Date: Thu, 20 Jun 2024 12:35:45 -0400 Subject: [PATCH] Apply linter It was previously not fully initialized --- sec_edgar_downloader/_orchestrator.py | 6 +++++- tests/test_orchestrator.py | 1 + 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/sec_edgar_downloader/_orchestrator.py b/sec_edgar_downloader/_orchestrator.py index 57d3c5e..ca9f532 100644 --- a/sec_edgar_downloader/_orchestrator.py +++ b/sec_edgar_downloader/_orchestrator.py @@ -131,7 +131,11 @@ def fetch_and_save_filings(download_metadata: DownloadMetadata, user_agent: str) successfully_downloaded = 0 to_download = aggregate_filings_to_download(download_metadata, user_agent) if download_metadata.accession_numbers_to_skip is not None: - to_download = [td for td in to_download if td.accession_number not in download_metadata.accession_numbers_to_skip] + to_download = [ + td + for td in to_download + if td.accession_number not in download_metadata.accession_numbers_to_skip + ] for td in to_download: try: diff --git a/tests/test_orchestrator.py b/tests/test_orchestrator.py index c9b6a48..34ff520 100644 --- a/tests/test_orchestrator.py +++ b/tests/test_orchestrator.py @@ -353,6 +353,7 @@ def test_fetch_and_save_filings_given_accession_numbers_to_skip( assert mock_download_filing.call_count == 1 assert mock_save_document.call_count == 1 + def test_fetch_and_save_filings_given_exception(user_agent, form_10k, apple_cik): limit = 2 download_metadata = DownloadMetadata(