fix: Migrate from attr.dataclass to dataclasses.dataclass (#207)

hipagesgroup · Sep 21, 2023 · fc02536 · fc02536
1 parent 4be2ed3
commit fc02536
Show file tree

Hide file tree

Showing 10 changed files with 10 additions and 10 deletions.
diff --git a/hip_data_tools/etl/adwords_to_athena.py b/hip_data_tools/etl/adwords_to_athena.py
@@ -3,7 +3,7 @@
 """
 from typing import List, Any, Optional, Tuple
 
-from attr import dataclass
+from dataclasses import dataclass
 from pandas import DataFrame
 
 from hip_data_tools.common import LOG

diff --git a/hip_data_tools/etl/adwords_to_s3.py b/hip_data_tools/etl/adwords_to_s3.py
@@ -5,7 +5,7 @@
 
 import numpy as np
 import pandas as pd
-from attr import dataclass
+from dataclasses import dataclass
 from googleads.adwords import ServiceQueryBuilder, ReportQuery
 from pandas import DataFrame
 

diff --git a/hip_data_tools/etl/athena_to_athena.py b/hip_data_tools/etl/athena_to_athena.py
@@ -3,7 +3,7 @@
 """
 from typing import Optional, List
 
-from attr import dataclass
+from dataclasses import dataclass
 
 from hip_data_tools.aws.athena import AthenaUtil
 from hip_data_tools.aws.common import AwsConnectionSettings, AwsConnectionManager

diff --git a/hip_data_tools/etl/athena_to_cassandra.py b/hip_data_tools/etl/athena_to_cassandra.py
@@ -2,7 +2,7 @@
 handle ETL of data from Athena to Cassandra
 """
 from typing import Optional, List
-from attr import dataclass
+from dataclasses import dataclass
 
 from hip_data_tools.apache.cassandra import CassandraConnectionSettings
 from hip_data_tools.aws.athena import AthenaUtil

diff --git a/hip_data_tools/etl/athena_to_dataframe.py b/hip_data_tools/etl/athena_to_dataframe.py
@@ -1,7 +1,7 @@
 """
 handle ETL of data from Athena to Pandas Dataframe
 """
-from attr import dataclass
+from dataclasses import dataclass
 
 from hip_data_tools.aws.athena import AthenaUtil
 from hip_data_tools.aws.common import AwsConnectionSettings, AwsConnectionManager

diff --git a/hip_data_tools/etl/athena_to_s3.py b/hip_data_tools/etl/athena_to_s3.py
@@ -3,7 +3,7 @@
 """
 from typing import Optional, List
 
-from attr import dataclass
+from dataclasses import dataclass
 
 from hip_data_tools.aws.athena import AthenaUtil
 from hip_data_tools.aws.common import AwsConnectionSettings, AwsConnectionManager

diff --git a/hip_data_tools/etl/google_sheet_to_athena.py b/hip_data_tools/etl/google_sheet_to_athena.py
@@ -2,7 +2,7 @@
 Module to deal with data transfer from Google sheets to Athena
 """
 
-from attr import dataclass
+from dataclasses import dataclass
 
 from hip_data_tools.aws.athena import AthenaUtil, get_table_settings_for_dataframe
 from hip_data_tools.aws.common import AwsConnectionManager

diff --git a/hip_data_tools/etl/google_sheet_to_s3.py b/hip_data_tools/etl/google_sheet_to_s3.py
@@ -2,7 +2,7 @@
 Module to deal with data transfer from Google sheets to S3
 """
 
-from attr import dataclass
+from dataclasses import dataclass
 
 from hip_data_tools.aws.common import AwsConnectionManager
 from hip_data_tools.aws.common import AwsConnectionSettings

diff --git a/hip_data_tools/etl/s3_to_cassandra.py b/hip_data_tools/etl/s3_to_cassandra.py
@@ -3,7 +3,7 @@
 """
 from typing import Optional, List
 
-from attr import dataclass
+from dataclasses import dataclass
 from cassandra.datastax.graph import Result
 from cassandra import ConsistencyLevel
 

diff --git a/hip_data_tools/google/common.py b/hip_data_tools/google/common.py
@@ -5,7 +5,7 @@
 from abc import ABC, abstractmethod
 from typing import List
 
-from attr import dataclass
+from dataclasses import dataclass
 from oauth2client.service_account import ServiceAccountCredentials
 
 from hip_data_tools.common import SecretsManager, ENVIRONMENT, KeyValueSource