[POT] Add quantization templates to wheel (#10557)

openvinotoolkit · Feb 22, 2022 · 1891967 · 1891967
1 parent 33062be
commit 1891967
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tools/pot/setup.py b/tools/pot/setup.py
@@ -25,6 +25,11 @@ def run(self):
             # install requires
             self.do_egg_install()
 
+        def_quant_path = os.path.join("configs", "default_quantization_template.json")
+        aa_quant_path = os.path.join("configs", "accuracy_aware_quantization_template.json")
+        copyfile(def_quant_path, os.path.join(self.install_purelib, prefix, "pot", def_quant_path))
+        copyfile(aa_quant_path, os.path.join(self.install_purelib, prefix, "pot", aa_quant_path))
+
         version_txt = os.path.join(prefix, "pot", "version.txt")
         if os.path.exists(version_txt):
             copyfile(os.path.join(version_txt),