Treesitter highlight messes up colors in imports

[elvisbejko]

The treesitter highlight doesn't integrate well with python.

I get the imports all highlighted with the same color (reddish):

Looking at similar code here on github, the expected look should be something like:

My github-nvim-theme configuration .lua and treesitter.lua follow:
treesitter.lua

return {
	"nvim-treesitter/nvim-treesitter",
	build = ":TSUpdate",
	config = function()
		local config = require("nvim-treesitter.configs")
		config.setup({
			auto_install = true,
			ensure_installed = { "lua", "c", "python", "rust", "latex" },
			highlight = { enable = true },
			indent = { enable = true },
		})
	end,
}

github-nvim-theme.lua

return {
	"projekt0n/github-nvim-theme",
	lazy = false,
	priority = 1000,
	config = function()
		require("github-theme").setup({
			groups = {
				all = {
					Normal = { bg = "NONE" },
					NormalNC = { bg = "NONE" },
					NetrwTreeBar = { bg = "NONE" }, -- Transparent background for netrw sidebar
					NetrwDir = { bg = "NONE" }, -- Transparent background for directories in netrw
					ColorColumn = { bg = "#1d1d1d" },
				},
			},
		})
		vim.cmd("colorscheme github_dark")
		vim.o.cursorline = true
		vim.api.nvim_set_hl(0, "CursorLineNr", { fg = "#FFF9D6", bold = true })
		vim.cmd("hi clear cursorline")
	end,
}

Since I recently bumped to a higher nvim version, I checked with earlier versions <0.10.x & the issue still persists.
Removing the highlight from treesitter.lua reverts the style to something too simplistic.