including plotly.graph_objects in environments

matteocacciola · matteocacciola · commit a342d1743077 · 2025-06-16T19:47:15.000+02:00
diff --git a/pandasai/core/code_execution/environment.py b/pandasai/core/code_execution/environment.py
@@ -30,6 +30,7 @@ def get_environment() -> dict:
         "plt": import_dependency("matplotlib.pyplot"),
         "np": import_dependency("numpy"),
         "px": import_dependency("plotly.express"),
+        "go": import_dependency("plotly.graph_objects"),
     }
 
     return env

Original file line number	Diff line number	Diff line change
`@@ -30,6 +30,7 @@ def get_environment() -> dict:`
`30`	`30`	`"plt": import_dependency("matplotlib.pyplot"),`
`31`	`31`	`"np": import_dependency("numpy"),`
`32`	`32`	`"px": import_dependency("plotly.express"),`
	`33`	`+ "go": import_dependency("plotly.graph_objects"),`
`33`	`34`	`}`
`34`	`35`
`35`	`36`	`return env`