Habitat handle no supported gpu

skyline-vscode/react-ui/src/App.js

@@ -103,6 +103,10 @@ function App() {
       if (event.data["message_type"] === "connection") {
         setConnectionStatus(event.data["status"]);
       } else if (event.data["message_type"] === "analysis") {
+        if(event.data.habitat[0][0]==="unavailable" && event.data.habitat[0][1]===-1.0){
+          event.data.habitat = profiling_data.habitat
+          event.data.habitat.push(["demo",1])
+        }
         processAnalysisState(event.data);
       } else if (event.data["message_type"] === "text_change") {
         setTextChanged(true);

skyline-vscode/react-ui/src/sections/Habitat.js

@@ -7,6 +7,26 @@ import Card from "react-bootstrap/Card";
 import { HabitatScatterGraph } from "../components/ScatterGraph";
 
 export default function Habitat({ habitatData }) {
+  const habitatIsDemo = habitatData.find(
+    (item) => item[0] === "demo" && item[1] === 1
+  );
+  if (habitatIsDemo) {
+    return (
+      <>
+        <div className="innpv-memory innpv-subpanel">
+          <Subheader icon="database">Habitat</Subheader>
+          <div className="innpv-subpanel-content">
+            <Container fluid className="mt-2">
+              <Row className="justify-content-md-center">
+                <h6>The local GPU is not supported by Habitat</h6>
+              </Row>
+            </Container>
+          </div>
+        </div>
+      </>
+    );
+  }
+
   return (
     <>
       <div className="innpv-memory innpv-subpanel">

skyline-vscode/react-ui/src/sections/Habitat.test.js

@@ -18,6 +18,22 @@ const data = [
   ["RTX4000", 20.2342],
 ];
 
+const noHabitatData = [
+  ["source", 22.029312],
+  ["P100",14.069682],
+  ["P4000", 127.268085], // 27.268085
+  ["RTX2070", 16.088268],
+  ["RTX2080Ti", 11.826558],
+  ["T4", 22.029312],
+  ["V100", 10.182922],
+  ["A100", 10.068596],
+  ["RTX3090", 9.841998],
+  ["A40", 11.558072],
+  ["A4000", 14.67059],
+  ["RTX4000", 20.2342],
+  ["demo",1]
+]
+
 beforeEach(() => {
   delete window.ResizeObserver;
   window.ResizeObserver = jest.fn().mockImplementation(() => ({
@@ -61,3 +77,10 @@ test("Shows graph when habitat data is present", () => {
   expect(container.querySelector('.recharts-responsive-container')).toBeTruthy();  // eslint-disable-line
 
 });
+
+test("no habitat data received from backend", ()=>{
+  // ARRANGE
+  const { container } = render(<Habitat habitatData={noHabitatData}/>);
+  // ASSERT
+  expect(screen.getByText(/the local gpu is not supported by habitat/i)).toBeTruthy();
+})

skyline-vscode/react-ui/src/sections/ProviderPanel.js

@@ -13,6 +13,7 @@ import Col from "react-bootstrap/Col";
 import Image from "react-bootstrap/Image";
 import Table from "react-bootstrap/Table";
 import Form from "react-bootstrap/Form";
+import Alert from 'react-bootstrap/Alert';
 
 import {
   cloudInstances,
@@ -68,6 +69,8 @@ const ProviderPanel = ({ numIterations, habitatData }) => {
     estimated_time: 0,
   });
 
+  const habitatIsDemo = habitatData.find(item=>item[0]==="demo" && item[1] === 1)
+
   const MAX_GPU = [1, 2, 4, 0]; // 0 is all
 
   const initial_data = populate_initial_data(habitatData);
@@ -175,6 +178,11 @@ const ProviderPanel = ({ numIterations, habitatData }) => {
       <div className="innpv-memory innpv-subpanel">
         <Subheader icon="database">Providers</Subheader>
         <Container fluid>
+          {habitatIsDemo && (
+            <Row>
+              <Alert variant="danger">Currently showing a demo data because local GPU is not supported by Habitat</Alert>
+            </Row>
+          )}
           <Row>
             <Col xl={12} xxl={8}>
               <Row>

skyline-vscode/react-ui/src/sections/ProviderPanel.test.js

@@ -19,6 +19,22 @@ const data = [
   ["RTX4000", 20.2342],
 ];
 
+const noHabitatData = [
+  ["source", 22.029312],
+  ["P100",14.069682],
+  ["P4000", 127.268085], // 27.268085
+  ["RTX2070", 16.088268],
+  ["RTX2080Ti", 11.826558],
+  ["T4", 22.029312],
+  ["V100", 10.182922],
+  ["A100", 10.068596],
+  ["RTX3090", 9.841998],
+  ["A40", 11.558072],
+  ["A4000", 14.67059],
+  ["RTX4000", 20.2342],
+  ["demo",1]
+];
+
 beforeEach(() => {
   delete window.ResizeObserver;
   window.ResizeObserver = jest.fn().mockImplementation(() => ({
@@ -62,3 +78,10 @@ test("Shows a scatter chart", () => {
   ).toBeTruthy();
   expect(screen.getByText(/select a configuration/i)).toBeTruthy();
 });
+
+test("no habitat data received from backend", ()=>{
+  // ARRANGE
+  const { container } = render(<ProviderPanel numIterations={numIterations} habitatData={noHabitatData} />);
+  // ASSERT
+  expect(screen.getByText(/currently showing a demo data because local gpu is not supported by habitat/i)).toBeTruthy();
+})