Merge pull request #21 from JuliaComputing/as/fix-clock-inference

fix: minor fixes for clock inference

Author: AayushSabharwal

lib/ModelingToolkitTearing/src/clock_inference/clock_inference.jl

@@ -199,6 +199,10 @@ function infer_clocks!(ci::ClockInference)
                 unshifted = args[1]
                 add_edge!(inference_graph, (
                     ClockVertex.Variable(i), ClockVertex.Variable(var_to_idx[unshifted])))
+                td = get_time_domain(unshifted)
+                if td isa TimeDomain
+                    add_edge!(inference_graph, (ClockVertex.Variable(i), ClockVertex.Clock(td)))
+                end
             end
             _ => nothing
         end

lib/ModelingToolkitTearing/src/clock_inference/interface.jl

@@ -34,10 +34,6 @@ function mark_discrete(state::StateSelection.TransformationState)
     unimplemented()
 end
 
-function postprocess_clock_inference(ci::ClockInference{TearingState}, state::TearingState)
-    substitute_sample_time(ci, state)
-end
-
 function error_sample_time(eq)
     error("$eq\ncontains `SampleTime` but it is not an Inferred discrete equation.")
 end