Python SDK release v0.6.2

tuneinsight · Jan 26, 2024 · abb3754 · abb3754
1 parent 9819af6
commit abb3754
Show file tree

Hide file tree

Showing 152 changed files with 9,544 additions and 5,740 deletions.
diff --git a/PKG-INFO b/PKG-INFO
@@ -1,10 +1,10 @@
 Metadata-Version: 2.1
 Name: tuneinsight
-Version: 0.6.0
-Summary: Diapason is the official Python SDK for the Tune Insight API. Version 0.6.0 targets the API v0.7.8.
+Version: 0.6.2
+Summary: Diapason is the official Python SDK for the Tune Insight API. Version 0.6.2 targets the API v0.8.0.
 License: Apache-2.0
 Author: Tune Insight SA
-Requires-Python: >=3.7.1
+Requires-Python: >=3.8,<3.12
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.8
@@ -13,20 +13,16 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Requires-Dist: PyYAML (>=6.0,<7.0)
 Requires-Dist: attrs (>=21.3.0)
+Requires-Dist: black (>=24.1.0,<25.0.0)
 Requires-Dist: certifi (>=2023.7.22,<2024.0.0)
-Requires-Dist: docker (>=6.0.1,<7.0.0)
 Requires-Dist: httpx (>=0.15.4,<0.24.0)
 Requires-Dist: matplotlib (>=3.5.0,<4.0.0)
 Requires-Dist: notebook (>=6.4.11,<7.0.0)
 Requires-Dist: pandas (>=1.3.5,<2.0.0)
-Requires-Dist: pylint (>=2.13.2,<3.0.0)
 Requires-Dist: python-dateutil (>=2.8.0,<3.0.0)
 Requires-Dist: python-dotenv (>=0.21.0,<0.22.0)
 Requires-Dist: python-keycloak (>=0.27.0,<0.28.0)
-Requires-Dist: pyvcf3 (>=1.0.3,<2.0.0)
-Requires-Dist: selenium (>=4.9.1,<5.0.0)
 Requires-Dist: typing-extensions (>=4.6.3,<5.0.0)
-Requires-Dist: wheel (>=0.38.1,<0.39.0)
 Description-Content-Type: text/markdown
 
 # Tune Insight Python SDK

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "tuneinsight"
-version = "0.6.0"
-description = "Diapason is the official Python SDK for the Tune Insight API. Version 0.6.0 targets the API v0.7.8."
+version = "0.6.2"
+description = "Diapason is the official Python SDK for the Tune Insight API. Version 0.6.2 targets the API v0.8.0."
 authors = ["Tune Insight SA"]
 license = "Apache-2.0"
 include = [
@@ -12,28 +12,37 @@ include = [
 readme = "src/tuneinsight/README.md"
 
 [tool.poetry.dependencies]
-python = ">= 3.7.1"
+python = ">= 3.8,<3.12"
 python-keycloak = "^0.27.0"
 pandas = "^1.3.5"
 PyYAML = "^6.0"
-#cloudpickle = "^2.0.0" // TODO: to add in the future to run python on backend
-wheel = "^0.38.1"
-pylint = "^2.13.2"
-docker = "^6.0.1"
 notebook = "^6.4.11"
 python-dotenv = "^0.21.0"
-pyvcf3 = "^1.0.3"
 python-dateutil = "^2.8.0"
 matplotlib = "^3.5.0"
-selenium = "^4.9.1"
 typing-extensions = "^4.6.3"
 
 # Required by ge_co_rest_api
 httpx = ">=0.15.4,<0.24.0"
 attrs = ">=21.3.0"
 certifi = "^2023.7.22"
+black = "^24.1.0"
 
+[tool.poetry.group.dev.dependencies]
+selenium = "^4.9.1"
+wheel = "^0.38.1"
+docker = "^6.0.1"
+pylint = "^2.13.2"
+pyvcf3 = "^1.0.3" # For GWAS .vcf file parsing
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"
+
+[tool.black]
+include = '\.pyi?$'
+extend-exclude='''
+(
+    src/tuneinsight/api
+)
+'''
diff --git a/src/tuneinsight/api/sdk/api/api_computations/compute.py b/src/tuneinsight/api/sdk/api/api_computations/compute.py
@@ -14,6 +14,7 @@
 from ...models.distributed_join import DistributedJoin
 from ...models.dummy import Dummy
 from ...models.encrypted_aggregation import EncryptedAggregation
+from ...models.encrypted_mean import EncryptedMean
 from ...models.encrypted_prediction import EncryptedPrediction
 from ...models.encrypted_regression import EncryptedRegression
 from ...models.error import Error
@@ -45,6 +46,7 @@ def _get_kwargs(
         "DistributedJoin",
         "Dummy",
         "EncryptedAggregation",
+        "EncryptedMean",
         "EncryptedPrediction",
         "EncryptedRegression",
         "GWAS",
@@ -135,6 +137,9 @@ def _get_kwargs(
     elif isinstance(json_body, PrivateSearch):
         json_json_body = json_body.to_dict()
 
+    elif isinstance(json_body, PrivateSearchSetup):
+        json_json_body = json_body.to_dict()
+
     else:
         json_json_body = json_body.to_dict()
 
@@ -196,6 +201,7 @@ def sync_detailed(
         "DistributedJoin",
         "Dummy",
         "EncryptedAggregation",
+        "EncryptedMean",
         "EncryptedPrediction",
         "EncryptedRegression",
         "GWAS",
@@ -218,9 +224,9 @@ def sync_detailed(
     Args:
         json_body (Union['AggregatedDatasetLength', 'Bootstrap', 'CollectiveKeyGen',
             'CollectiveKeySwitch', 'DatasetStatistics', 'DistributedJoin', 'Dummy',
-            'EncryptedAggregation', 'EncryptedPrediction', 'EncryptedRegression', 'GWAS', 'HybridFL',
-            'KeySwitchedComputation', 'PrivateSearch', 'PrivateSearchSetup', 'RelinKeyGen',
-            'RotKeyGen', 'SampleExtraction', 'SetIntersection', 'SetupSession',
+            'EncryptedAggregation', 'EncryptedMean', 'EncryptedPrediction', 'EncryptedRegression',
+            'GWAS', 'HybridFL', 'KeySwitchedComputation', 'PrivateSearch', 'PrivateSearchSetup',
+            'RelinKeyGen', 'RotKeyGen', 'SampleExtraction', 'SetIntersection', 'SetupSession',
             'StatisticalAggregation', 'SurvivalAggregation', 'VBinnedAggregation']):
 
     Raises:
@@ -256,6 +262,7 @@ def sync(
         "DistributedJoin",
         "Dummy",
         "EncryptedAggregation",
+        "EncryptedMean",
         "EncryptedPrediction",
         "EncryptedRegression",
         "GWAS",
@@ -278,9 +285,9 @@ def sync(
     Args:
         json_body (Union['AggregatedDatasetLength', 'Bootstrap', 'CollectiveKeyGen',
             'CollectiveKeySwitch', 'DatasetStatistics', 'DistributedJoin', 'Dummy',
-            'EncryptedAggregation', 'EncryptedPrediction', 'EncryptedRegression', 'GWAS', 'HybridFL',
-            'KeySwitchedComputation', 'PrivateSearch', 'PrivateSearchSetup', 'RelinKeyGen',
-            'RotKeyGen', 'SampleExtraction', 'SetIntersection', 'SetupSession',
+            'EncryptedAggregation', 'EncryptedMean', 'EncryptedPrediction', 'EncryptedRegression',
+            'GWAS', 'HybridFL', 'KeySwitchedComputation', 'PrivateSearch', 'PrivateSearchSetup',
+            'RelinKeyGen', 'RotKeyGen', 'SampleExtraction', 'SetIntersection', 'SetupSession',
             'StatisticalAggregation', 'SurvivalAggregation', 'VBinnedAggregation']):
 
     Raises:
@@ -309,6 +316,7 @@ async def asyncio_detailed(
         "DistributedJoin",
         "Dummy",
         "EncryptedAggregation",
+        "EncryptedMean",
         "EncryptedPrediction",
         "EncryptedRegression",
         "GWAS",
@@ -331,9 +339,9 @@ async def asyncio_detailed(
     Args:
         json_body (Union['AggregatedDatasetLength', 'Bootstrap', 'CollectiveKeyGen',
             'CollectiveKeySwitch', 'DatasetStatistics', 'DistributedJoin', 'Dummy',
-            'EncryptedAggregation', 'EncryptedPrediction', 'EncryptedRegression', 'GWAS', 'HybridFL',
-            'KeySwitchedComputation', 'PrivateSearch', 'PrivateSearchSetup', 'RelinKeyGen',
-            'RotKeyGen', 'SampleExtraction', 'SetIntersection', 'SetupSession',
+            'EncryptedAggregation', 'EncryptedMean', 'EncryptedPrediction', 'EncryptedRegression',
+            'GWAS', 'HybridFL', 'KeySwitchedComputation', 'PrivateSearch', 'PrivateSearchSetup',
+            'RelinKeyGen', 'RotKeyGen', 'SampleExtraction', 'SetIntersection', 'SetupSession',
             'StatisticalAggregation', 'SurvivalAggregation', 'VBinnedAggregation']):
 
     Raises:
@@ -367,6 +375,7 @@ async def asyncio(
         "DistributedJoin",
         "Dummy",
         "EncryptedAggregation",
+        "EncryptedMean",
         "EncryptedPrediction",
         "EncryptedRegression",
         "GWAS",
@@ -389,9 +398,9 @@ async def asyncio(
     Args:
         json_body (Union['AggregatedDatasetLength', 'Bootstrap', 'CollectiveKeyGen',
             'CollectiveKeySwitch', 'DatasetStatistics', 'DistributedJoin', 'Dummy',
-            'EncryptedAggregation', 'EncryptedPrediction', 'EncryptedRegression', 'GWAS', 'HybridFL',
-            'KeySwitchedComputation', 'PrivateSearch', 'PrivateSearchSetup', 'RelinKeyGen',
-            'RotKeyGen', 'SampleExtraction', 'SetIntersection', 'SetupSession',
+            'EncryptedAggregation', 'EncryptedMean', 'EncryptedPrediction', 'EncryptedRegression',
+            'GWAS', 'HybridFL', 'KeySwitchedComputation', 'PrivateSearch', 'PrivateSearchSetup',
+            'RelinKeyGen', 'RotKeyGen', 'SampleExtraction', 'SetIntersection', 'SetupSession',
             'StatisticalAggregation', 'SurvivalAggregation', 'VBinnedAggregation']):
 
     Raises: