Code

Tackle Typing and Linting Errors#379

Code

Merged

slinderman merged 29 commits intomainprobml/dynamax:mainfrom ghd/typingprobml/dynamax:ghd/typingCopy head branch name to clipboardJan 27, 2025

+1,057-707

Commits

Commits on Oct 16, 2024

• Add ruff ignore F722

  gileshd

  committed
  3f89961

  Copy full SHA for 3f89961

• Remove unused imports

  gileshd

  committed
  3ad1fa7

  Copy full SHA for 3ad1fa7

• Convert strings with latex to raw-strings

  Show description for dc7f675
  gileshd

  committed
  dc7f675

  Copy full SHA for dc7f675

• Prepend space in uni-dim jaxtyping hints

  Show description for 20a1df7
  gileshd

  committed
  20a1df7

  Copy full SHA for 20a1df7

• Fix jr.PRNGKey type hints

  gileshd

  committed
  ca9a75c

  Copy full SHA for ca9a75c

• Rename and change PRNGKey Type

  Show description for 75205f8
  gileshd

  committed
  75205f8

  Copy full SHA for 75205f8

• Add IntScalar type

  gileshd

  committed
  42c5c22

  Copy full SHA for 42c5c22

• Minor arg and type changes in utils/utils.py

  Show description for 7da57ef
  gileshd

  committed
  7da57ef

  Copy full SHA for 7da57ef

• Update HMM[Parameter|Property]Set protocols

  Show description for 62df3ba
  gileshd

  committed
  62df3ba

  Copy full SHA for 62df3ba

• Update type annotations in hmm base classes

  gileshd

  committed
  45134e8

  Copy full SHA for 45134e8

• Update type annotations in hmm inference code.

  Show description for 0d4ca74
  gileshd

  committed
  0d4ca74

  Copy full SHA for 0d4ca74

• Fix type annotations in hmm parallel inference

  gileshd

  committed
  cfd3bfc

  Copy full SHA for cfd3bfc

• Add further type annotations to hmm transitions class

  Show description for d0a0e0f
  gileshd

  committed
  d0a0e0f

  Copy full SHA for d0a0e0f

• Add further type annotations to hmm initial base class

  Show description for 4b2c40b
  gileshd

  committed
  4b2c40b

  Copy full SHA for 4b2c40b

• Add further type annotations to categorical hmm

  gileshd

  committed
  4c84ddc

  Copy full SHA for 4c84ddc

• Add further type annotations to arhmm

  gileshd

  committed
  7e04c9e

  Copy full SHA for 7e04c9e

• Add further type annotations to linreghmm

  gileshd

  committed
  b2fffa4

  Copy full SHA for b2fffa4

• Add further type annotations to Bernoulli HMM

  gileshd

  committed
  a45e61b

  Copy full SHA for a45e61b

• Add further type annotations to Gamma HMM

  gileshd

  committed
  32db62a

  Copy full SHA for 32db62a

• Add further type annotations to Gaussian HMMs

  Show description for 81138f8
  gileshd

  committed
  81138f8

  Copy full SHA for 81138f8

• Add further type annotations to gmhmms

  Show description for 4e36520
  gileshd

  committed
  4e36520

  Copy full SHA for 4e36520

• Add further type annotations to logreg hmm

  gileshd

  committed
  4b7ee53

  Copy full SHA for 4b7ee53

• Add further type annotations to multinomialhmm

  gileshd

  committed
  e246606

  Copy full SHA for e246606

• Add further type annotations to poisson hmm

  gileshd

  committed
  e19748d

  Copy full SHA for e19748d

• Add further type annotations to categorical glm hmm

  gileshd

  committed
  c2098cf

  Copy full SHA for c2098cf

• Fix LinearGaussianSSM.sample type hint

  gileshd

  committed
  547c610

  Copy full SHA for 547c610

• Change type hints to jaxtyping in slds code

  gileshd

  committed
  58c9127

  Copy full SHA for 58c9127

Commits on Jan 27, 2025

• Merge branch 'main' into ghd/typing

  slinderman

  authored
  2b251bc

  Copy full SHA for 2b251bc

• scaling down weights for poisson. apparently an upstream change in jax or tfp led to some numerical instabilities with the Poisson LDS with log link function.

  slinderman

  committed
  1d8f831

  Copy full SHA for 1d8f831