Disable torchtext warning

helicalAI · May 22, 2024 · fcaf499 · fcaf499
1 parent cf1fb93
commit fcaf499
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/helical/models/scgpt/tokenizer/gene_tokenizer.py b/helical/models/scgpt/tokenizer/gene_tokenizer.py
@@ -8,6 +8,8 @@
 import numpy as np
 import pandas as pd
 import torch
+import torchtext 
+torchtext.disable_torchtext_deprecation_warning()
 import torchtext.vocab as torch_vocab
 from torchtext.vocab import Vocab