Upstreaming fbsync changes (#1367)

pytorch · Jul 28, 2021 · 760a625 · 760a625
1 parent c57b1fb
commit 760a625
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/torchtext/utils.py b/torchtext/utils.py
@@ -1,3 +1,4 @@
+import torch
 import csv
 import hashlib
 import os
@@ -241,3 +242,10 @@ def extract_archive(from_path, to_path=None, overwrite=False):
     else:
         raise NotImplementedError(
             "We currently only support tar.gz, .tgz, .gz and zip achives.")
+
+
+def _log_class_usage(klass):
+    identifier = "torchtext"
+    if klass and hasattr(klass, "__name__"):
+        identifier += f".{klass.__name__}"
+    torch._C._log_api_usage_once(identifier)
diff --git a/torchtext/vocab/vocab.py b/torchtext/vocab/vocab.py
@@ -1,6 +1,7 @@
 import torch
 import torch.nn as nn
 from typing import Dict, List, Optional
+from torchtext.utils import _log_class_usage
 
 
 class Vocab(nn.Module):
@@ -14,7 +15,7 @@ class Vocab(nn.Module):
     def __init__(self, vocab):
         super(Vocab, self).__init__()
         self.vocab = vocab
-        torch._C._log_api_usage_once(f"torchtext.{self.__class__.__name__}")
+        _log_class_usage(__class__)
 
     @property
     def is_jitable(self):