utils.py 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. # Copyright NeMo (https://github.com/NVIDIA/NeMo). All Rights Reserved.
  2. #
  3. # Licensed under the Apache License, Version 2.0 (the "License");
  4. # you may not use this file except in compliance with the License.
  5. # You may obtain a copy of the License at
  6. #
  7. # http://www.apache.org/licenses/LICENSE-2.0
  8. #
  9. # Unless required by applicable law or agreed to in writing, software
  10. # distributed under the License is distributed on an "AS IS" BASIS,
  11. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. # See the License for the specific language governing permissions and
  13. # limitations under the License.
  14. import csv
  15. import os
  16. def get_abs_path(rel_path):
  17. """
  18. Get absolute path
  19. Args:
  20. rel_path: relative path to this file
  21. Returns absolute path
  22. """
  23. return os.path.dirname(os.path.abspath(__file__)) + '/' + rel_path
  24. def load_labels(abs_path):
  25. """
  26. loads relative path file as dictionary
  27. Args:
  28. abs_path: absolute path
  29. Returns dictionary of mappings
  30. """
  31. label_tsv = open(abs_path)
  32. labels = list(csv.reader(label_tsv, delimiter="\t"))
  33. return labels