thomasaarholt

# print stashes that exist for the currently switched-to branch
GREEN='\033[0;32m'
NC='\033[0m' # No Color

branch=$(git rev-parse --abbrev-ref HEAD)
stashes=`git stash list | grep "WIP on $branch"`
if [ "$stashes" ]
then
  echo "${GREEN}You have the following stashes for this branch:"
  echo "${stashes}${NC}"

thomasaarholt

import sys
from time import time

N_repeats = 100


def func(foo: list[int]):
    return [x for x in foo if x % 2 == 0]

thomasaarholt

# Use the pymc master branch
from collections.abc import Callable
from typing import Literal

from numpy._typing import ArrayLike
import pymc as pm
from numpy import ndarray
from pymc.distributions.shape_utils import (
    Dims,
    Shape,

thomasaarholt

function stash() {
    # Check if stash is empty
    if ! git stash list &>/dev/null; then
        echo "No stashes found."
        return
    fi

    # Fetch relative dates and messages from git stash
    local relative_dates=$(git stash list --format="%ar")
    local messages=$(git stash list --format="%gs" | sed 's/^On //; s/^WIP on //')

thomasaarholt

import json
import polars as pl

def dtype_to_json(dtype: pl.DataType) -> str:
    return json.dumps(str(dtype))


def json_to_dtype(json_dtype_str: str) -> pl.DataType:
    from polars.datatypes.classes import (  # noqa F401
        Array,

thomasaarholt

[[language]]
name = "python"
scope = "source.python"
injection-regex = "python"
file-types = ["py","pyi","py3","pyw",".pythonstartup",".pythonrc"]
shebangs = ["python"]
roots = [".", "pyproject.toml", "pyrightconfig.json"]
comment-token = "#"
language-servers = ["pyright", "ruff"]
indent = { tab-width = 4, unit = "    " }

thomasaarholt

from typing import Any
import polars as pl
import json

from pydantic import BaseModel

original_expression = (pl.col("foo") * 2) == pl.col("bar")


class Column(BaseModel, arbitrary_types_allowed=True):

thomasaarholt

[[language]]
name = "python"
scope = "source.python"
injection-regex = "python"
file-types = ["py","pyi","py3","pyw",".pythonstartup",".pythonrc"]
shebangs = ["python"]
roots = [".", "pyproject.toml", "pyrightconfig.json"]
comment-token = "#"
language-servers = ["pyright", "ruff"]
indent = { tab-width = 4, unit = "    " }

thomasaarholt

from typing import Any
import json
import structlog
import pandas as pd # pip install pandas
import polars as pl # pip install polars
from pydantic import BaseModel

class PydanticModel(BaseModel):
    a: int
    b: str

thomasaarholt

[[language]]
name = "python"
scope = "source.python"
injection-regex = "python"
file-types = ["py","pyi","py3","pyw",".pythonstartup",".pythonrc"]
shebangs = ["python"]
roots = [".", "pyproject.toml", "pyrightconfig.json"]
comment-token = "#"
language-servers = ["pyright", "ruff"]
indent = { tab-width = 4, unit = "    " }