Skip to content

Commit 0234761

Browse files
committed
Server: minor model.py data_source_by_id name update
1 parent c8ea654 commit 0234761

File tree

1 file changed

+3
-3
lines changed
  • src/server/endpoints/covidcast_utils

1 file changed

+3
-3
lines changed

src/server/endpoints/covidcast_utils/model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,7 @@ def _load_data_sources():
202202

203203

204204
data_sources, data_sources_df = _load_data_sources()
205-
data_source_by_id = {d.source: d for d in data_sources}
205+
data_sources_by_id = {d.source: d for d in data_sources}
206206

207207

208208
def _load_data_signals(sources: List[DataSource]):
@@ -231,7 +231,7 @@ def _load_data_signals(sources: List[DataSource]):
231231
data_signals_by_key = {d.key: d for d in data_signals}
232232
# also add the resolved signal version to the signal lookup
233233
for d in data_signals:
234-
source = data_source_by_id.get(d.source)
234+
source = data_sources_by_id.get(d.source)
235235
if source and source.uses_db_alias:
236236
data_signals_by_key[(source.db_source, d.signal)] = d
237237

@@ -261,7 +261,7 @@ def create_source_signal_alias_mapper(source_signals: List[SourceSignalSet]) ->
261261
alias_to_data_sources: Dict[str, List[DataSource]] = {}
262262
transformed_sets: List[SourceSignalSet] = []
263263
for ssset in source_signals:
264-
source = data_source_by_id.get(ssset.source)
264+
source = data_sources_by_id.get(ssset.source)
265265
if not source or not source.uses_db_alias:
266266
transformed_sets.append(ssset)
267267
continue

0 commit comments

Comments
 (0)