hestia-earth-utils 0.16.16__py3-none-any.whl → 0.16.18__py3-none-any.whl

hestia_earth/utils/pipeline.py

@@ -1,3 +1,4 @@
+import os
 from os.path import join
 import json
 import numpy as np
@@ -375,3 +376,29 @@ def get_related_nodes_data(
         }
         for node in related_nodes
     ]
+
+
+def start_instances(multiple_instances: bool = False):
+    import boto3
+
+    client = boto3.client("ec2")
+    instance_ids = os.getenv("START_INSTANCE_ID").split(",")
+    ids = instance_ids if multiple_instances else [instance_ids[0]]
+    try:
+        for instance_id in ids:
+            response = client.describe_instances(InstanceIds=[instance_id])
+            state = response["Reservations"][0]["Instances"][0]["State"]["Name"]
+            print(f"Checking {instance_id}: Current state is '{state}'")
+
+            if state in ["pending", "running"]:
+                print(f" -> Instance {instance_id} is already active. Skipping.")
+                continue
+
+            elif state == "stopped":
+                print(f" -> Instance {instance_id} is stopped. Starting it now...")
+                client.start_instances(InstanceIds=[instance_id])
+                break
+    except client.exceptions.ClientError as e:
+        print(f"Error starting instances: {str(e)}")
+
+    print("Could not start any instance")

hestia_earth/utils/pivot/pivot_json.py

@@ -51,6 +51,13 @@ def _combine_node_ids(nodes: list):
     return {"@id": ";".join([node.get("@id", node.get("id")) for node in nodes])}
 
 
+_COMPACT_JSON_KEYS = ["boundary", "distribution", "covarianceMatrix"]
+
+
+def _combine_json_compact(key: str, value: dict):
+    return json.dumps(value).replace(" ", "") if key in _COMPACT_JSON_KEYS else value
+
+
 def _base_pivoted_value(key: str, value, is_top_level: bool):
     # handle list of Nodes
     return (
@@ -58,13 +65,13 @@ def _base_pivoted_value(key: str, value, is_top_level: bool):
             _combine_node_ids(value)
             if isinstance(value[0], dict) and value[0].get("@type") in NODE_TYPES
             else (
-                json.dumps(value, separators=(",", ":"))
-                if any([is_top_level, key in ["distribution"]])
+                _combine_json_compact(key, value)
+                if any([is_top_level, key in _COMPACT_JSON_KEYS])
                 else value
             )
         )
         if isinstance(value, list)
-        else value
+        else _combine_json_compact(key, value)
     )
 
 

hestia_earth/utils/version.py

@@ -1 +1 @@
-VERSION = "0.16.16"
+VERSION = "0.16.18"

{hestia_earth_utils-0.16.16.dist-info → hestia_earth_utils-0.16.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hestia_earth_utils
-Version: 0.16.16
+Version: 0.16.18
 Summary: HESTIA's utils library
 Home-page: https://gitlab.com/hestia-earth/hestia-utils
 Author: HESTIA Team

{hestia_earth_utils-0.16.16.dist-info → hestia_earth_utils-0.16.18.dist-info}/RECORD

@@ -9,25 +9,25 @@ hestia_earth/utils/emission.py,sha256=rHHf5vwe-RxTOaOJ9N0MuyJOLpDnPjjxv6MHYSpPgc
 hestia_earth/utils/lookup.py,sha256=SArKqjqs_Yt5cC6TIk12WVQNToun2spZ_iNUm4fX9FA,8274
 hestia_earth/utils/lookup_utils.py,sha256=cjYIFSQpcdSCoHqCoz3f5Odkpv_2j5aGn0COOIdz3hg,6948
 hestia_earth/utils/model.py,sha256=YzQ4Cs3HBlhLGNDvrfv2pM_FrNZhZkg9h7JKIH3CHFE,4626
-hestia_earth/utils/pipeline.py,sha256=LYxPXhuSWTwmsLRz_nl_JscFgGYhUAbNQx4Zla5_REM,10292
+hestia_earth/utils/pipeline.py,sha256=EvZKSCle3bFNfy1slD24-4dcziiFuW-e_3Re2liaYao,11319
 hestia_earth/utils/request.py,sha256=EfitmS13abPxfl5gSAOyHRUYhk4R2Rfv94fF9lvOT00,723
 hestia_earth/utils/stats.py,sha256=vTNyKcMKmX0DoodM9QEG7HF8qm2Wf-4ckMWQFWZ1VgE,34729
 hestia_earth/utils/table.py,sha256=MOJDo5fQPRDogAty_UXbO9-EXFwz97m0f7--mOM17lQ,2363
 hestia_earth/utils/term.py,sha256=aBVYuYv55nPqJPyt5mN4Fz652s_1hwUPckNUZX0pMP8,1064
 hestia_earth/utils/tools.py,sha256=WMx05cBtBR8mYQnLLBA2cgF1x2tI41514diFDeX4gLQ,7533
-hestia_earth/utils/version.py,sha256=SGKvksJDyfirCQnk3wbuotlbC2CvazXTR8lJe6kBC6U,20
+hestia_earth/utils/version.py,sha256=GRD_WbP1hGteFSYD0I4Y_A4gXGnyzTXUqqTJ4jRXspM,20
 hestia_earth/utils/pivot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hestia_earth/utils/pivot/_shared.py,sha256=ZBmZRLbDN2JHODvQAh9jH_61Koh7Ts0C52xmiw_Bhpo,2827
 hestia_earth/utils/pivot/pivot_csv.py,sha256=1zGYuC_xlrLoX5FYSG3mt5HSc03CO-0p8J0oAmyI1VI,11778
-hestia_earth/utils/pivot/pivot_json.py,sha256=yWbsuFeTQJPk4H2b7ibVALGJa3qfKGyNuPeqM8yrcsI,11030
+hestia_earth/utils/pivot/pivot_json.py,sha256=bqR3SmoHu9-Ij4UtUEbsZjEqtqq1jdS4Nx3CpyLN3Mc,11263
 hestia_earth/utils/storage/__init__.py,sha256=uNX6_EHWWnNUIm4Ng7L43-cQmuc6NGFAxXye85saIXQ,922
 hestia_earth/utils/storage/_azure_client.py,sha256=mseexhzjteRDzzoFe2fEXe9MYLmvjG3sfl9-eha4ZW0,1353
 hestia_earth/utils/storage/_local_client.py,sha256=KbYqTfniIU5R5J1m_unCQip9kOz9EGIGI0OH0QvD8eo,551
 hestia_earth/utils/storage/_s3_client.py,sha256=8TCxiHfxE7G8kdp3CnEFrxgmPwfPyci3-blsowE2T7o,3146
 hestia_earth/utils/storage/_sns_client.py,sha256=pvtXYw-sQ8ns3mlDz7ld9iZp3FYSm0xSXSXMJ5IPnBc,380
-hestia_earth_utils-0.16.16.data/scripts/hestia-format-upload,sha256=IhLAHHPJqRgUcht-M_EUEsRMbRbMfshig07o488zscM,703
-hestia_earth_utils-0.16.16.data/scripts/hestia-pivot-csv,sha256=0YBuGuyPO8rytod6iwWEKiQdSlr9JLuD001k6U5t6no,1163
-hestia_earth_utils-0.16.16.dist-info/METADATA,sha256=iOf4uwvgxtoZgFdSjW3uElhZfRf9vRQFJL_LC9ojvlM,1873
-hestia_earth_utils-0.16.16.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
-hestia_earth_utils-0.16.16.dist-info/top_level.txt,sha256=q0QxKEYx9uLpAD5ZtC7Ypq29smEPfOzEAn7Xv8XHGOQ,13
-hestia_earth_utils-0.16.16.dist-info/RECORD,,
+hestia_earth_utils-0.16.18.data/scripts/hestia-format-upload,sha256=IhLAHHPJqRgUcht-M_EUEsRMbRbMfshig07o488zscM,703
+hestia_earth_utils-0.16.18.data/scripts/hestia-pivot-csv,sha256=0YBuGuyPO8rytod6iwWEKiQdSlr9JLuD001k6U5t6no,1163
+hestia_earth_utils-0.16.18.dist-info/METADATA,sha256=p-XSakIkjWkzvUUUU2dgegRFRWBm0ip0sfINgSBI7aM,1873
+hestia_earth_utils-0.16.18.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+hestia_earth_utils-0.16.18.dist-info/top_level.txt,sha256=q0QxKEYx9uLpAD5ZtC7Ypq29smEPfOzEAn7Xv8XHGOQ,13
+hestia_earth_utils-0.16.18.dist-info/RECORD,,

{hestia_earth_utils-0.16.16.dist-info → hestia_earth_utils-0.16.18.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.10.1)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any