From 9ad4a62ac15096b5d035696d7fe763be6e0657a1 Mon Sep 17 00:00:00 2001
From: Eric Yu <chenyangyu@instagram.com>
Date: Fri, 13 Nov 2020 15:16:47 -0800
Subject: [PATCH] add py support for textray_details.thrift

Summary: add py support for textray_details.thrift, which will be used by PyText module

Reviewed By: mikekgfb

Differential Revision: D24964672

fbshipit-source-id: 835f988c4f652baeeb8546e8e0e074f9c60f212d
---
 pytext/torchscript/module.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/pytext/torchscript/module.py b/pytext/torchscript/module.py
index a22a8edb3..495df6a4d 100644
--- a/pytext/torchscript/module.py
+++ b/pytext/torchscript/module.py
@@ -3,6 +3,10 @@
 from typing import Dict, List, Optional, Tuple
 
 import torch
+from multiray.textray.textray_details.ttypes import (  # noqa
+    QuantizationSchema,
+    FeatureSchema,
+)
 from pytext.torchscript.batchutils import (
     max_tokens,
     make_prediction_texts,