Skip to content

Commit

Permalink
formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
varunagrawal committed Dec 10, 2024
1 parent a98ac0f commit 8145086
Show file tree
Hide file tree
Showing 5 changed files with 12 additions and 7 deletions.
4 changes: 2 additions & 2 deletions gtsam/discrete/discrete.i
Original file line number Diff line number Diff line change
Expand Up @@ -61,14 +61,14 @@ virtual class DecisionTreeFactor : gtsam::DiscreteFactor {
DecisionTreeFactor(const std::vector<gtsam::DiscreteKey>& keys, string table);

DecisionTreeFactor(const gtsam::DiscreteConditional& c);

void print(string s = "DecisionTreeFactor\n",
const gtsam::KeyFormatter& keyFormatter =
gtsam::DefaultKeyFormatter) const;
bool equals(const gtsam::DecisionTreeFactor& other, double tol = 1e-9) const;

size_t cardinality(gtsam::Key j) const;

double operator()(const gtsam::DiscreteValues& values) const;
gtsam::DecisionTreeFactor operator*(const gtsam::DecisionTreeFactor& f) const;
size_t cardinality(gtsam::Key j) const;
Expand Down
3 changes: 2 additions & 1 deletion python/gtsam/tests/test_DecisionTreeFactor.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,9 +13,10 @@

import unittest

from gtsam.utils.test_case import GtsamTestCase

from gtsam import (DecisionTreeFactor, DiscreteDistribution, DiscreteValues,
Ordering)
from gtsam.utils.test_case import GtsamTestCase


class TestDecisionTreeFactor(GtsamTestCase):
Expand Down
4 changes: 2 additions & 2 deletions python/gtsam/tests/test_DiscreteBayesTree.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,8 +19,8 @@

import gtsam
from gtsam import (DiscreteBayesNet, DiscreteBayesTreeClique,
DiscreteConditional, DiscreteFactorGraph,
DiscreteValues, Ordering)
DiscreteConditional, DiscreteFactorGraph, DiscreteValues,
Ordering)


class TestDiscreteBayesNet(GtsamTestCase):
Expand Down
3 changes: 2 additions & 1 deletion python/gtsam/tests/test_DiscreteConditional.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,9 +13,10 @@

import unittest

from gtsam import DecisionTreeFactor, DiscreteConditional, DiscreteKeys
from gtsam.utils.test_case import GtsamTestCase

from gtsam import DecisionTreeFactor, DiscreteConditional, DiscreteKeys

# Some DiscreteKeys for binary variables:
A = 0, 2
B = 1, 2
Expand Down
5 changes: 4 additions & 1 deletion python/gtsam/tests/test_DiscreteFactorGraph.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,9 +14,12 @@
import unittest

import numpy as np
from gtsam import DecisionTreeFactor, DiscreteConditional, DiscreteFactorGraph, DiscreteKeys, DiscreteValues, Ordering, Symbol
from gtsam.utils.test_case import GtsamTestCase

from gtsam import (DecisionTreeFactor, DiscreteConditional,
DiscreteFactorGraph, DiscreteKeys, DiscreteValues, Ordering,
Symbol)

OrderingType = Ordering.OrderingType


Expand Down

0 comments on commit 8145086

Please sign in to comment.