"""Routines used for extracting the raw NREL ATB data."""fromdagsterimportOutput,assetfrompudl.extract.extractorimportGenericMetadata,raw_df_factoryfrompudl.extract.parquetimportParquetExtractor
[docs]classExtractor(ParquetExtractor):"""Extractor for NREL ATB."""def__init__(self,*args,**kwargs):"""Initialize the module. Args: ds (:class:datastore.Datastore): Initialized datastore. """self.METADATA=GenericMetadata("nrelatb")super().__init__(*args,**kwargs)
[docs]defraw_nrelatb__data(raw_nrelatb__all_dfs):"""Extract raw NREL ATB data from annual parquet files to one dataframe. Returns: An extracted NREL ATB dataframe. """returnOutput(value=raw_nrelatb__all_dfs["data"])