From bf509ffed17fea851989077de0f002a5734bddc5 Mon Sep 17 00:00:00 2001
From: Jeffrey Aven <javen@avensolutions.com>
Date: Wed, 17 Jul 2024 15:11:13 +1000
Subject: [PATCH] provider updates

---
 Dockerfile              | 2 +-
 notebooks/aws/aws.ipynb | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 3c6a294..30bd06e 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -10,7 +10,7 @@ USER stackql
 RUN stackql exec 'registry pull aws' || (echo "Failed to pull aws provider" && exit 1)
 RUN stackql exec 'registry pull google' || (echo "Failed to pull google provider" && exit 1)
 RUN stackql exec 'registry pull github' || (echo "Failed to pull github provider" && exit 1)
-# RUN stackql exec 'registry pull azure'
+RUN stackql exec 'registry pull azure' || (echo "Failed to pull azure provider" && exit 1)
 # RUN stackql exec 'registry pull k8s'
 # RUN stackql exec 'registry pull netlify'
 # RUN stackql exec 'registry pull okta'
diff --git a/notebooks/aws/aws.ipynb b/notebooks/aws/aws.ipynb
index 33f657e..dd326fb 100644
--- a/notebooks/aws/aws.ipynb
+++ b/notebooks/aws/aws.ipynb
@@ -27,10 +27,10 @@
    "source": [
     "%%stackql\n",
     "SELECT \n",
-    "state, \n",
+    "JSON_EXTRACT(state, '$$.Name') as instance_state, \n",
     "count(*) as num_instances\n",
     "FROM aws.ec2.instances \n",
-    "WHERE region IN ($regions)\n",
+    "WHERE region = '$region'\n",
     "GROUP BY state"
    ]
   },
@@ -40,7 +40,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "_.plot(kind='bar', title='Instances by State', x='state', y='num_instances')"
+    "_.plot(kind='bar', title='Instances by State', x='instance_state', y='num_instances')"
    ]
   },
   {