Skip to content

Commit

Permalink
Fix dependency issue at OSS.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 674439698
  • Loading branch information
bdu91 authored and LIT team committed Sep 13, 2024
1 parent 362cf16 commit ee34f2d
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion lit_nlp/app.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@
# ==============================================================================
"""LIT backend, as a standard WSGI app."""

import collections
from collections.abc import Callable, Iterable, Mapping, Sequence
import functools
import glob
Expand Down Expand Up @@ -1021,7 +1022,7 @@ def __init__(
def get_dataset_specs(self) -> dict[str, dict[str, str]]:
datasets_with_spec = collections.defaultdict(dict)
for name, ds in self._datasets.items():
for field_name, lit_data_class in ds.spec().items():
for field_name, lit_data_class in ds.spec().items():
datasets_with_spec[name][field_name] = type(lit_data_class).__name__
return datasets_with_spec

Expand Down

0 comments on commit ee34f2d

Please sign in to comment.