fix typing issue in python 3.8

marimo/_server/api/set_cell_config.py

@@ -2,6 +2,7 @@
 from __future__ import annotations
 
 from dataclasses import dataclass
+from typing import Dict
 
 import tornado.web
 
@@ -14,7 +15,7 @@
 @dataclass
 class SetCellConfig:
     # Map from Cell ID to (possibily partial) CellConfig
-    configs: dict[CellId_t, dict[str, object]]
+    configs: Dict[CellId_t, Dict[str, object]]
 
 
 class SetCellConfigHandler(tornado.web.RequestHandler):

marimo/_server/api/test_model.py

@@ -5,7 +5,9 @@
 from dataclasses import asdict, dataclass
 from typing import Any, Dict, List, Tuple
 
+from marimo._server import api
 from marimo._server.api.model import parse_raw
+from marimo._server.api.set_cell_config import SetCellConfig
 
 
 @dataclass
@@ -114,3 +116,13 @@ class Nested:
 
         parsed = parse_raw(serialize(nested), Nested)
         assert parsed == nested
+
+    def test_set_cell_config(self) -> None:
+        config = SetCellConfig(
+            {"0": {"disabled": True}, "1": {"disabled": False}}
+        )
+        parsed = parse_raw(
+            serialize(config), api.set_cell_config.SetCellConfig
+        )
+        print(parsed)
+        assert parsed == config

marimo/_smoke_tests/altair_charts.py

@@ -1,3 +1,4 @@
+# Copyright 2023 Marimo. All rights reserved.
 import marimo
 
 __generated_with = "0.1.24"