Skip to content

Commit 5b998d5

Browse files
committed
Server: minor model.py data_source_by_id name update
1 parent eac044f commit 5b998d5

File tree

1 file changed

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

1 file changed

+3
-4
lines changed

src/server/endpoints/covidcast_utils/model.py

Lines changed: 3 additions & 4 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,12 +231,11 @@ 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

238238

239-
240239
def get_related_signals(signal: DataSignal) -> List[DataSignal]:
241240
return [s for s in data_signals if s != signal and s.signal_basename == signal.signal_basename]
242241

@@ -266,7 +265,7 @@ def create_source_signal_alias_mapper(source_signals: List[SourceSignalPair]) ->
266265
alias_to_data_sources: Dict[str, List[DataSource]] = {}
267266
transformed_pairs: List[SourceSignalPair] = []
268267
for pair in source_signals:
269-
source = data_source_by_id.get(pair.source)
268+
source = data_sources_by_id.get(pair.source)
270269
if not source or not source.uses_db_alias:
271270
transformed_pairs.append(pair)
272271
continue

0 commit comments

Comments
 (0)