[POT]: Requirements clean up (#14630)

* Cleaned up requirements in POT

* Fixed issue

* Removed pandas usage from POT

* Update setup.py

* Removed the upperbound for numpy in the case of Python < 3.11

* Removed numpy logic
This commit is contained in:
Alexander Kozlov 2023-01-09 13:44:09 +04:00 committed by GitHub
parent c20c867683
commit 893da35826
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 4 additions and 7 deletions

View File

@ -9,7 +9,6 @@ from functools import partial
from pathlib import Path
import numpy as np
import pandas as pd
import scipy.optimize
from ..qnoise_estimator.algorithm import QuantNoiseEstimator
@ -166,7 +165,7 @@ class OptimizationAlgorithm(Algorithm):
Path('/'.join(self._result_filename.split('/')[:-1])).mkdir(
parents=True, exist_ok=True
)
pd.DataFrame(self._results).to_csv(self._result_filename)
np.savetxt(self._result_filename, self._results, delimiter=",", fmt='%s')
if self._dump_model_prefix:
dump_path = self._dump_model_prefix + '{:05}'.format(
self._optimization_iter

View File

@ -4,7 +4,6 @@
from copy import deepcopy
import numpy as np
import pandas as pd
from openvino.tools.mo.graph.graph import rename_node
from .utils import get_composite_model
@ -98,7 +97,7 @@ class QuantNoiseEstimator(Algorithm):
'layer_name': list(stat_calculation_layers.values()),
}
if 'results_dump_filename' in self._config:
pd.DataFrame(noise_data).to_csv(self._config['results_dump_filename'])
np.savetxt(self._config['results_dump_filename'], noise_data, delimiter=",", fmt='%s')
return noise_data
def layerwise_fq_noise(self, model):
@ -175,7 +174,7 @@ class QuantNoiseEstimator(Algorithm):
noise_data = {'noise_metric': qnoise_values, 'layer_name': node_names}
if 'results_dump_filename' in self._config:
pd.DataFrame(noise_data).to_csv(self._config['results_dump_filename'])
np.savetxt(self._config['results_dump_filename'], noise_data, delimiter=",", fmt='%s')
return noise_data
def get_nonquantized_model(self, model):

View File

@ -65,15 +65,14 @@ if '--install-dev-extras' in sys.argv:
sys.argv.remove('--install-dev-extras')
INSTALL_REQUIRES = [
"numpy>=1.16.6",
"scipy~=1.7; python_version == '3.7'",
"scipy>=1.8; python_version >= '3.8'",
"jstyleson>=0.0.2",
"numpy>=1.16.6",
"addict>=2.4.0",
"networkx<=2.8.8",
"tqdm>=4.54.1",
"texttable>=1.6.3",
"pandas~=1.3.5",
"openvino-telemetry>=2022.1.0"
]